RuntimeError: exp_vml_cpu not implemented for 'Long'
其他
2020-01-24 18:57:32
阅读次数: 0
问题描述: 执行torch.exp(torch.tensor([2]))报错, RuntimeError: exp_vml_cpu not implemented for ‘Long’
原因: torch.exp()操作不支持Long类型的张量作为输入
解决方法: 将张量转为浮点型即可, 执行torch.exp(torch.tensor([2.0]))或者执行torch.exp(torch.tensor([2],dtype=torch.float))
总结: pytorch中的很多操作不支持Long类型的张量, 所以会出现这一类报错信息:RuntimeError: xxx_vml_cpu not implemented for ‘Long’, 只需要把输入的张量改成浮点类型即可
发布了489 篇原创文章 ·
获赞 101 ·
访问量 15万+
转载自blog.csdn.net/littlehaes/article/details/103837338
