tensorly.base
.partial_fold¶

partial_fold
(unfolded, mode, shape, skip_begin=1, skip_end=0)[source]¶ Refolds a partially unfolded tensor
 Parameters
 unfoldedndarray
a partially unfolded tensor
 modeint
indexing starts at 0, therefore mode is in range(0, tensor.ndim)
 shapetuple
the shape of the original full tensor (including skipped dimensions)
 skip_beginint, optional, default is 1
number of dimensions to leave untouched at the beginning
 skip_endint, optional
number of dimensions to leave untouched at the end
 Returns
 ndarray
partially refolded tensor