ott.neural.networks.potentials.PotentialTrainState.replace
==========================================================

.. currentmodule:: ott.neural.networks.potentials

.. automethod:: PotentialTrainState.replace