Torch.Tensor.softmax missing from documentation

Site Feedback
1

Tensors have a softmax() method, but it’s not displayed here: “torch.Tensor — PyTorch master documentation” torch.Tensor — PyTorch master documentation

Is there a reason for this or was it overlooked?

2

Are you referring to Softmax — PyTorch 1.9.0 documentation ?

3

Yes, it’s the same function but as a method for a Tensor.

These 2 calls are identical:

torch.nn.Softmax(1)(tensor)

and

tensor.softmax(1)

but the Tensor variant isn’t listed on the documentation anywhere (at least not that I could find).

4

Thanks for raising this issue. Would you mind creating an issue on GitHub so that we can track and fix it? Would you also be interested in fixing the missing docs?