ott.solvers.nn.neuraldual.NeuralDualSolver.get\_step\_fn ======================================================== .. currentmodule:: ott.solvers.nn.neuraldual .. automethod:: NeuralDualSolver.get_step_fn