PyTorch Forums
Confusion with nn.CrossEntropyLoss
ptrblck
March 22, 2023, 9:53am
2
This example code
might be helpful which shows how the target is used to index the logits.
1 Like
show post in topic