SparseCategoricalCrossEntropyLoss

constructor(dim: Int = -1)(source)

Parameters

dim

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