博主用nn.softmax时出错“softmax_lastdim_kernel_impl“ not implemented for ‘Long‘

仅作为记录，大佬请跳过。

要进行softmax的tensor里面的元素是 float

解决

将int改为float即可：

参考

感谢大佬博主文章：传送门

tensor的nn.softmax示例

import torch
import torch.nn as nn

_a=torch.tensor([float(1),float(1),float(0)])
m=nn.Softmax(dim=0)
_a_nnsoftmax=m(_a)
print(_a,'\n',_a_nnsoftmax)