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

22 lines
338 B
ReStructuredText
Raw Normal View History

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
2023-12-22 14:13:41 +08:00
cross_entropy
kl_div_loss
2023-12-18 05:23:03 +08:00
l1_loss
mse_loss
nll_loss
2023-12-22 14:13:41 +08:00
smooth_l1_loss
2024-01-04 12:14:05 +08:00
triplet_loss
hinge_loss
huber_loss
log_cosh_loss