Skip to main content

Browse the archive

Revision d001f75585f36015df36feafd1f5ad6c1618b8dc authored by TUNA Caglayan on 03 May 2021, 07:22:28 UTC, committed by TUNA Caglayan on 12 May 2021, 12:26:22 UTC 
fix tensor_to_vec
1 parent d38084d
Raw File
_factorized_tensor.py 
from . import backend as T
from . import tenalg
from collections.abc import Mapping
from abc import ABCMeta

class FactorizedTensor(Mapping, metaclass=ABCMeta):
    """Base Class for Tensors in Factorized form
    """
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)

    def to_tensor(self):
        return NotImplementedError
    
    def to_unfolded(self, mode):
        return NotImplementedError
    
    def to_vec(self):
        return NotImplementedError

    def norm(self):
        """Norm l2 of the tensor"""
        return T.norm(self.to_tensor())

    def mode_dot(self, matrix_or_tensor, mode):
        return tenalg.mode_dot(self.to_tensor(), matrix_or_tensor, mode)
back to top