nn.softmax pour le classificateur de sconvoutionnel pur

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