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.
