Rate this Page

torch.softmax#

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

Alias fortorch.nn.functional.softmax().