pytorch softmax dim=-1

对于输入是2D tensor，参数dim = -1 和 dim = 1 都是对每一行求softmax
因为可以通过 0，1去索引 两个维度，也可以分别通过 -2，-1去索引这两个对应的维度。于是-1和1对应

import torch.nn.functional as F
F.softmax(some_2D_tensor, dim=-1)

测试：https://www.cnblogs.com/jeshy/p/10933882.html