ott.neural.networks.potentials.PotentialTrainState.potential\_gradient\_fn
==========================================================================

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

.. autoattribute:: PotentialTrainState.potential_gradient_fn