ott.geometry.costs.Bures.norm#

Bures.norm(x)[source]#

Compute norm of Gaussian, sq. 2-norm of mean + trace of covariance.

Parameters

x (jax._src.numpy.ndarray.ndarray) –