tensorly.base.partial_fold

partial_fold(unfolded, mode, shape, skip_begin=1, skip_end=0)[source]

Re-folds a partially unfolded tensor

Parameters:	unfolded : ndarray a partially unfolded tensor mode : int indexing starts at 0, therefore mode is in range(0, tensor.ndim) shape : tuple the shape of the original full tensor (including skipped dimensions) skip_begin : int, optional, default is 1 number of dimensions to leave untouched at the beginning skip_end : int, optional number of dimensions to leave untouched at the end
Returns:	ndarray partially re-folded tensor