ColossalAI/docs/colossalai/colossalai.nn.loss.cross_en...

6 lines
142 B
ReStructuredText

colossalai.nn.loss.cross\_entropy\_1d
=====================================
.. automodule:: colossalai.nn.loss.cross_entropy_1d
:members: