Implicit dimension choice for softmax warning

10

https://github.com/pytorch/pytorch/blob/master/torch/nn/functional.py#L814.

For matrices, it’s 1. For others, it’s 0.

3 Likes