colossalai.nn.loss
==================

.. automodule:: colossalai.nn.loss
   :members:


.. toctree::
   :maxdepth: 2

   colossalai.nn.loss.base_loss
   colossalai.nn.loss.cross_entropy_1d
   colossalai.nn.loss.cross_entropy_2d
   colossalai.nn.loss.cross_entropy_2p5d
   colossalai.nn.loss.cross_entropy_3d