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
class Module
class Object
trait Matchable
class Any