mlx/docs/build/html/_sources/python/nn/losses.rst

.. _losses:

.. currentmodule:: mlx.nn.losses

Loss Functions
--------------

.. autosummary::
   :toctree: _autosummary_functions
   :template: nn-module-template.rst

   binary_cross_entropy
   cross_entropy
   kl_div_loss
   l1_loss
   mse_loss
   nll_loss
   smooth_l1_loss
   triplet_loss
docs 2023-12-18 05:23:03 +08:00			`.. _losses:`

			`.. currentmodule:: mlx.nn.losses`

			`Loss Functions`
			`--------------`

			`.. autosummary::`
			`:toctree: _autosummary_functions`
			`:template: nn-module-template.rst`

			`binary_cross_entropy`
docs 2023-12-22 14:13:41 +08:00			`cross_entropy`
			`kl_div_loss`
docs 2023-12-18 05:23:03 +08:00			`l1_loss`
			`mse_loss`
			`nll_loss`
docs 2023-12-22 14:13:41 +08:00			`smooth_l1_loss`
			`triplet_loss`