-
Notifications
You must be signed in to change notification settings - Fork 16
Closed
Labels
enhancementNew feature or requestNew feature or request
Milestone
Description
cov currently expects an array with at most two dimensions. Given a single (m, n), it produces one (m, m).
It would be useful if cov were vectorized operate on arrays of arbitrary dimensionality as a batch. Given a batch of samples as an array with shape batch_shape + (m, n), it would produce a batch of covariance matrices as an array of shape batch_shape + (m, m).
This would be consistent with other standard functions like matmul. matrix_transpose, and many linalg extension functions.
(I'd have an immediate use for this vectorizing scipy.stats.epps_singleton_2samp.)
Metadata
Metadata
Assignees
Labels
enhancementNew feature or requestNew feature or request