pytorch-softmax

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
input=torch.rand(5,3,2)
out0=F.softmax(input,0)
out1=F.softmax(input,1)
out2=F.softmax(input,2)
print(out0)
print(out1)
print(out2)

结果分析

out0对应维度大小为5	out1对应维度大小为3	out2对应维度为2
0.1971+0.2582+0.1640 +0.1454+0.2352=0.9999	0.4020+0.2787+0.3193=1	0.4588+0.5412=1