torch.nn.loss
package torch.nn.loss
Members list
Type members
Classlikes
final class CrossEntropyLoss extends Module
This criterion computes the cross entropy loss between input and target.
This criterion computes the cross entropy loss between input and target.
Attributes
- Source
- CrossEntropyLoss.scala
- Supertypes
In this article