title: convert nn.Softmax to torch.tensor
body: <p>I have a softmax function at the end of my Neural Network.
  I want to have the probabilities as torch.tensor.For it I am using <code> torch.tensor(nn.softmax(x))</code> and getting error <code>RuntimeError: Could not infer dtype of Softmax</code>.</p>
  <p>May I please know what I am doing wrong here or is there any other way to do it.</p>