Trouble getting probability from softmax

13

Any plans on its depreciation similar to nn.functional.sigmoid as mentioned here