“nn.softmax для чистого сцепления” Ответ

nn.softmax для чистого сцепления

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