tensorly.tucker_tensor.tucker_mode_dot

tucker_mode_dot(tucker_tensor, matrix_or_vector, mode, keep_dim=False, copy=False)[source]

n-mode product of a Tucker tensor and a matrix or vector at the specified mode

Parameters

tucker_tensor : tl.TuckerTensor or (core, factors)

matrix_or_vector : ndarray

1D or 2D array of shape (J, i_k) or (i_k, ) matrix or vectors to which to n-mode multiply the tensor

mode : int

Returns

TuckerTensor = (core, factors)

mode-mode product of tensor by matrix_or_vector * of shape (i_1, ..., i_{k-1}, J, i_{k+1}, ..., i_N) if matrix_or_vector is a matrix * of shape (i_1, ..., i_{k-1}, i_{k+1}, ..., i_N) if matrix_or_vector is a vector

See also

tucker_multi_mode_dot

chaining several mode_dot in one call