You are working on the data-apis/array-api-extra repository. The source code is checked out at /testbed at…
You are working on the data-apis/array-api-extra repository. The source code is checked out at /testbed at…: a task in SWE-Lego-RL-2699 (Harbor dataset). 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…
The task
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)`.
Part of Lego-X/Lego-RL-2699.