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.