ott.solvers.nn package#

Neural Dual#

neuraldual.NeuralDualSolver(input_dim[, ...])

Solver of the ICNN-based Kantorovich dual.

ICNN#

icnn.ICNN(dim_data, dim_hidden[, init_std, ...])

Input convex neural network (ICNN) architecture with initialization.

layers.PositiveDense(dim_hidden[, ...])

A linear transformation using a weight matrix with all entries positive.