spd_learn.functional.unvec_batch#

spd_learn.functional.unvec_batch(X_vec: Tensor, n: int) → Tensor[source]#

Unvectorizes a batch of tensors along the last dimension.

Parameters:

  • X_vec (torch.Tensor) – A batch of vectorized matrices with shape (…, n * k).

  • n (int) – The number of rows in the output matrices.

Returns:

A batch of matrices with shape (…, n, k).

Return type:

torch.Tensor