分类：dim参数

2021
08-10

Pytorch中torch.nn.Softmax的dim参数用法说明

Pytorch中torch.nn.Softmax的dim参数使用含义涉及到多维tensor时，对softmax的参数dim总是很迷，下面用一个例子说明importtorch.nnasnnm=nn.Softmax(dim=0)n=nn.Softmax(dim=1)k=nn.Softmax(dim=2)input=torch.randn(2,2,3)print(input)print(m(input))print(n(input))print(k(input))输出：inputtensor([[[0.5450,-0.6264,1.0446],[0.6324,1.9069,0.7158]],[[1.0092,0.2421,-0.8928],[0.0344,0.9723,0.4... 继续阅读 >