mirror of
https://github.com/ml-explore/mlx.git
synced 2025-06-26 02:33:21 +08:00
22 lines
338 B
ReStructuredText
22 lines
338 B
ReStructuredText
.. _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
|
|
hinge_loss
|
|
huber_loss
|
|
log_cosh_loss |