fold_views

Contents

fold_views#

stable_pretraining.data.fold_views(tensor, idx)[source]#

Fold a tensor containing multiple views back into separate views.

Parameters:
  • tensor – Tensor containing concatenated views

  • idx – Sample indices to determine view boundaries

Returns:

Tuple of tensors, one for each view