deepfold.losses.procrustes.unflatten_batch_dims

deepfold.losses.procrustes.unflatten_batch_dims(tensor: Tensor, batch_shape: Size) → Tensor