python代码A = torch.arange(20).reshape(5, 4) print(A.mean()) 出现报错RuntimeError: mean(): could not infer output dtype. Input dtype must be either a floating point or complex dtype. Got: Long 请问是为什么,如何解决
时间: 2023-06-10 11:05:17 浏览: 168
python torch.utils.data.DataLoader使用方法
5星 · 资源好评率100%
这个报错是因为 A 的数据类型是 Long 类型,而 mean 函数只支持浮点型或者复杂型数据类型。所以可以通过将 A 的数据类型转换为浮点型来解决这个问题,例如使用 A = torch.arange(20).reshape(5, 4).float() 来将 A 的数据类型转换为浮点型。这样就可以成功地调用 mean 函数了。
阅读全文