ott.core.neuraldual.NeuralDual#

class ott.core.neuraldual.NeuralDual(state_f, state_g)[source]#

Neural Kantorovich dual.

state_f#

optimal potential f

state_g#

optimal potential g

Methods

distance(source, target)

Given potentials f and g, compute the overall distance.

inverse_transport(data)

Transport source data samples with potential g.

transport(data)

Transport source data samples with potential g.

Attributes

f

g