torch.softmax

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

是 torch.nn.functional.softmax() 的別名。