- class ott.geometry.costs.Euclidean#
Note that the Euclidean distance is not cast as a
TICost, since this would correspond to \(h\) being
jax.numpy.linalg.norm(), whose gradient is not invertible, because the function is not strictly convex (it is linear on rays).
Compute matrix of all pairwise costs, including the
Compute matrix of all pairwise costs, excluding the
Compute Euclidean norm using custom jvp implementation.