import torch.nn.functional as F class torch.nn.Softmax(input, dim) 参数:dim: 指明维度:dim=0表示按列计算;dim=1表示按行计算。 Input: n维张量意义:对n维输入张量运用Softmax函数,将张量的每个元素缩放到(0,1)区间且和为1