ott.geometry.costs.UnbalancedBures.pairwise#

UnbalancedBures.pairwise(x, y)[source]#

Compute dot-product for unbalanced Bures.

Parameters

  • x (Array) – Array of shape [n_points + n_points + n_dim ** 2,] corresponding to the raveled mass, means and the covariance matrix.

  • y (Array) – Array of shape [n_points + n_points + n_dim ** 2,] corresponding to the raveled mass, means and the covariance matrix.

Return type

Array

Returns

The cost.