ott.neural.networks.potentials.PotentialTrainState.apply\_fn
============================================================

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

.. autoattribute:: PotentialTrainState.apply_fn