Search notes:

torch.nn.CrossEntropyLoss

torch.nn.CrossEntropyLoss is a criterion. It calculates the cross entropy loss between input logits and target.

See also

PyTorch example: Hello World with Fashion MNIST

Index