torch nn softmax code example

Example: nn.softmax for pure sconvoultional classifier

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