docs
0.0-2dfa388-SNAPSHOT
API
Generated with
docs/torch/torch.nn/torch.nn.loss/CrossEntropyLoss

CrossEntropyLoss

torch.nn.loss.CrossEntropyLoss
final class CrossEntropyLoss extends Module

This criterion computes the cross entropy loss between input and target.

Attributes

Source
CrossEntropyLoss.scala
Graph
Supertypes
class Module
class Object
trait Matchable
class Any

Members list

Value members

Concrete methods

def apply[D <: DType](input: Tensor[D], target: Tensor[_]): Tensor[D]

Attributes

Source
CrossEntropyLoss.scala
override def hasBias(): Boolean

Attributes

Definition Classes
Source
CrossEntropyLoss.scala

Inherited methods

def apply(fn: Module => Unit): Module.this.type

Attributes

Inherited from:
Module
Source
Module.scala
def eval(): Unit

Attributes

Inherited from:
Module
Source
Module.scala
def isTraining: Boolean

Attributes

Inherited from:
Module
Source
Module.scala
def load(inputArchive: InputArchive): Unit

Attributes

Inherited from:
Module
Source
Module.scala
def loadStateDict(stateDict: Map[String, Tensor[DType]]): Unit

Attributes

Inherited from:
Module
Source
Module.scala
def modules: Seq[Module]

Attributes

Inherited from:
Module
Source
Module.scala
def modules(recurse: Boolean): Seq[Module]

Attributes

Inherited from:
Module
Source
Module.scala
def namedBuffers(recurse: Boolean): SeqMap[String, Tensor[_]]

Attributes

Inherited from:
Module
Source
Module.scala
def namedChildren: SeqMap[String, Module]

Attributes

Inherited from:
Module
Source
Module.scala
def namedModules: SeqMap[String, Module]

Attributes

Inherited from:
Module
Source
Module.scala
def namedParameters(recurse: Boolean): SeqMap[String, Tensor[_]]

Attributes

Inherited from:
Module
Source
Module.scala
def parameters(recurse: Boolean): Seq[Tensor[_]]

Attributes

Inherited from:
Module
Source
Module.scala
def parameters: Seq[Tensor[_]]

Attributes

Inherited from:
Module
Source
Module.scala
def register[M <: Module](child: M, n: String)(using name: Name): M

Attributes

Inherited from:
Module
Source
Module.scala
def registerBuffer[D <: DType](name: String, tensor: Tensor[D]): Tensor[D]

Adds a buffer to the module.

Adds a buffer to the module.

Attributes

Inherited from:
Module
Source
Module.scala
def registerBuffer[D <: DType](t: Tensor[D], n: String)(using name: Name): Tensor[D]

Attributes

Inherited from:
Module
Source
Module.scala
def registerModule[M <: Module](child: M, n: String)(using name: Name): M

Attributes

Inherited from:
Module
Source
Module.scala
def registerParameter[D <: DType](t: Tensor[D], requiresGrad: Boolean, n: String)(using name: Name): Tensor[D]

Attributes

Inherited from:
Module
Source
Module.scala
def save(outputArchive: OutputArchive): Unit

Attributes

Inherited from:
Module
Source
Module.scala
def summarize: String

Attributes

Inherited from:
Module
Source
Module.scala
def to(device: Device): Module.this.type

Attributes

Inherited from:
Module
Source
Module.scala
override def toString(): String

Returns a string representation of the object.

Returns a string representation of the object.

The default representation is platform dependent.

Attributes

Returns

a string representation of the object.

Definition Classes
Module -> Any
Inherited from:
Module
Source
Module.scala
def train(on: Boolean): Unit

Attributes

Inherited from:
Module
Source
Module.scala
Generated with