分类：平方根报错

2021
07-01

PyTorch平方根报错的处理方案

问题描述初步使用PyTorch进行平方根计算，通过range()创建一个张量，然后对其求平方根。a=torch.tensor(list(range(9)))b=torch.sqrt(a)报出以下错误：RuntimeError:sqrt_vml_cpunotimplementedfor'Long'原因Long类型的数据不支持log对数运算,为什么Tensor是Long类型?因为创建List数组时默认使用的是int,所以从List转成torch.Tensor后,数据类型变成了Long。print(a.dtype)torch.int64解决方法提前将数据类型指定... 继续阅读 >