PyTorch平方根报错处理
问题描述
初步使用PyTorch进行平方根计算,通过range()创建一个张量,然后对其求平方根。
a = torch.tensor(list(range(9)))
b = torch.sqrt(a)
报出以下错误:
RuntimeError: sqrt_vml_cpu not implemented for 'Long'
原因
Long类型的数据不支持log对数运算, 为什么Tensor是Long类型? 因为创建List数组时默认使用的是int, 所以从List转成torch.Tensor后, 数据类型
原创
2020-08-08 14:10:53 ·
2214 阅读 ·
0 评论