spd_learn.functional.covariance

spd_learn.functional.covariance(input: Tensor) → Tensor

Computes the covariance matrix of multivariate data.

The input tensor is assumed to have shape (…, n_channels, n_times), where … represents any number of leading dimensions.

Parameters:

input (torch.Tensor) – Input tensor with shape (…, n_channels, n_times).

Returns:

Covariance matrices with shape (…, n_channels, n_channels).

Return type:

torch.Tensor