https://github.com/tensorly/tensorly
Revision 60bad71b9e684035e65575cef12c12143f6cc888 authored by Jean KOSSAIFI on 20 October 2016, 18:01:00 UTC, committed by Jean KOSSAIFI on 20 October 2016, 18:01:00 UTC
1 parent 7f9d41c
Raw File
Tip revision: 60bad71b9e684035e65575cef12c12143f6cc888 authored by Jean KOSSAIFI on 20 October 2016, 18:01:00 UTC
Tucker decomposition
Tip revision: 60bad71
_norm.py
import numpy as np

# Author: Jean Kossaifi


def norm(tensor, order):
    """Computes the l-order norm of tensor

    Parameters
    ----------
    tensor : ndarray
    order : int

    Returns
    -------
    float
        l-order norm of tensor
    """
    if order == 1:
        return np.sum(np.abs(tensor))
    elif order == 2:
        return np.sqrt(np.sum(tensor**2))
    else:
        return np.sum(np.abs(tensor)**order)**(1/order)
back to top