torch.softmax

torch.softmax(input, dim, *, dtype=None) → Tensor

torch.nn.functional.softmax() 的别名。