F.softmax函数dim解读

F.softmax(score, dim=1)

dim=1就是对score矩阵中 所有第1维下标不同，其他维下标均相同的元素进行操作（softmax）

比如a[0][8][15]和a[7][8][15]都会被选中，同时进行同一操作
而a[0][8][15]和a[0][8][15]不会
a[0][8][15]和a[0][7][15]也不会