spd_learn.functional.vec_batch#
- spd_learn.functional.vec_batch(X: Tensor) Tensor[source]#
Vectorizes a batch of tensors along the last two dimensions.
- Parameters:
X (torch.Tensor) – A batch of matrices with shape (…, n, k).
- Returns:
A batch of vectorized matrices with shape (…, n * k).
- Return type: