Skip to main content

Browse the archive

https://github.com/tensorly/tensorly
22 March 2024, 04:41:05 UTC
Revision 41ce484c66ca8d1690709439cee87feb3136860f authored by Jean Kossaifi on 06 January 2017, 16:52:58 UTC, committed by Jean Kossaifi on 06 January 2017, 16:52:58 UTC 
DOC: Left out file.
1 parent e06106f
Raw File
Tip revision: 41ce484c66ca8d1690709439cee87feb3136860f authored by Jean Kossaifi on 06 January 2017, 16:52:58 UTC
DOC: Left out file.
Tip revision: 41ce484
_higher_order_moment.py 
from ._khatri_rao import khatri_rao

# Author: Jean Kossaifi

# License: BSD 3 clause



def higher_order_moment(matrix, order=3):
    """Higher order moment of a matrix of observations

        Computes the higher moment of order `order` of `matrix`

        Each row of `matrix` represents a samples
        (i.e. an observation)

    Parameters
    ----------
    matrix : 2D-array
        array of shape (n_samples, n_features)

        i.e. each row is a sample
    order : int, optional
        order of the moment to compute
    """
    matrix = matrix - matrix.mean(axis=0)
    n_features = matrix.shape[-1]
    t = khatri_rao([matrix.T] * order).mean(axis=1)
    return t.reshape([n_features] * order)
back to top