0
点赞
收藏
分享

微信扫一扫

RuntimeError: log_vml_cpu not implemented for 'Long'


​​welcome to my blog​​

问题描述

执行​​torch.log(torch.from_numpy(np.array([1,2,2])))​​报错, 错误信息为:RuntimeError: log_vml_cpu not implemented for ‘Long’

原因

Long类型的数据不支持log对数运算, 为什么Tensor是Long类型? 因为创建numpy 数组时没有指定dtype, 默认使用的是int64, 所以从numpy array转成torch.Tensor后, 数据类型变成了Long

解决方法

提前将数据类型指定为浮点型, 重新执行​​torch.log(torch.from_numpy(np.array([1,2,2],np.float)))​​, 没有报错, 成功解决问题


举报

相关推荐

0 条评论