ott.geometry.costs.RegTICost.h_legendre

ott.geometry.costs.RegTICost.h_legendre#

RegTICost.h_legendre(z)[source]#

Legendre transform of h().

This function uses custom_vjp to apply Danskin’s theorem [Danskin, 1967] when differentiating.

Parameters:

z (Array) – Array of shape [d,].

Return type:

float

Returns:

The value.