The Pytorch documentation says: torch.nn.functional.softmax(input, dim=None, _stacklevel=3)
But when I run: torch.nn.functional.softmax(a, dim=1), I gen the following error: TypeError: softmax() got an unexpected keyword argument 'dim'
Is this a bug or am I doing something wrong?