SparseCategoricalCrossEntropyLoss
Parameters
dim
The dimension along which to compute the softmax. Default is -1 (last dimension).
The dimension along which to compute the softmax. Default is -1 (last dimension).