nn.softmax para um classificador puro sconvoulcional

>>> m = nn.Softmax(dim=1)
>>> input = torch.randn(2, 3)
>>> output = m(input)
Eager Elk