deepfold.losses.procrustes.flatten_batch_dims

deepfold.losses.procrustes.flatten_batch_dims(tensor: Tensor, end_dim: int) → Tuple[Tensor, Size]