How to avoid nan in softmax?

[smth]

you can use nn.LogSoftmax, it is numerically more stable and is less likely to nan than using Softmax