Class CrossEntropyLoss

Hierarchy

Loss
- CrossEntropyLoss

Index

Constructors

Methods

Constructors

constructor

new CrossEntropyLoss(reduction?: Reduction): CrossEntropyLoss
Parameters
- reduction: Reduction = 'mean'
Returns CrossEntropyLoss
Overrides Loss.constructor
- Defined in nn/loss.ts:68

Methods

forward

forward(input: Tensor, target: Tensor): Tensor
Parameters
- input: Tensor
- target: Tensor
Returns Tensor
Overrides Loss.forward
- Defined in nn/loss.ts:73