is this the same as the dim dimension in max (https://pytorch.org/docs/stable/generated/torch.max.html 17)?
Max doc (https://pytorch.org/docs/stable/generated/torch.max.html 17):
dim (int) – the dimension to reduce.

Softmax doc (https://pytorch.org/docs/master/generated/torch.nn.Softmax.html 4):
dim (int) – A dimension along which Softmax will be computed (so every slice along dim will sum to 1).

Basically does dim always mean the same thing?