pytorch-softmax torch.nn.functional.softmax() 一句话描述 将Tensor数值转为概率分布(取值0-1,所有取值之和为1)[1,1]->[0.5,0.5] 官方文档 其中,重点介绍参数dim-沿着某一维度进行计算。比较抽象可以看下面小例程。 小例程 import torch import torch.nn.functional as F