tensorly.base
.partial_fold
- partial_fold(unfolded, mode, shape, skip_begin=1, skip_end=0)[source]
Re-folds 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 re-folded tensor