What are criteria for choosing "dim=0 or 1" for nn.Softmax and nn.LogSoftmax

(rahul yadav) #1

Hi,

What are criteria for choosing “dim=0 or 1” for nn.Softmax and nn.LogSoftmax

0 Likes