ott.geometry.costs.UnbalancedBures.norm
ott.geometry.costs.UnbalancedBures.norm#
- UnbalancedBures.norm(x)[source]#
Compute norm of Gaussian for unbalanced Bures.
- Parameters
x (
Array) – Array of shape[n_points + n_points + n_dim ** 2,], potentially batched, corresponding to the raveled mass, means and the covariance matrix.- Return type
Array- Returns
The norm, array of shape
[]or[batch,]in the batched case.