ott.geometry.costs.SqPNorm#
- class ott.geometry.costs.SqPNorm(p)[source]#
Squared p-norm of the difference of two vectors.
Uses custom implementation of norm to avoid NaN values when differentiating the norm of x-x.
- Parameters:
p (
float
) – Power of the p-norm, \(\ge 1\).
Methods
all_pairs
(x, y)Compute matrix of all pairwise costs, including the
norms
.all_pairs_pairwise
(x, y)Compute matrix of all pairwise costs, excluding the
norms
.barycenter
(weights, xs)Barycentric operator.
h
(z)TI function acting on difference of \(x-y\) to output cost.
h_legendre
(z)Legendre transform of
h()
.pairwise
(x, y)Compute cost as evaluation of
h()
on \(x-y\).Attributes