RuntimeError: scatter(): Expected dtype int64 for index
RuntimeError: scatter(): Expected dtype int64 for index1.报错原因:scatter要求数据是int64类型,而我在定义tensor时写的是torch.Tensor(x),应该写成torch.LongTensor(x),指定为int64类型。2.解决方法找到原数据的定义方式,改!一般在dtype=np.int64;dtype=np.float3
·
RuntimeError: scatter(): Expected dtype int64 for index
1.报错原因:
scatter要求数据是int64类型,而我在定义tensor时写的是torch.Tensor(x),应该写成torch.LongTensor(x),指定为int64类型。
2.解决方法
找到原数据的定义方式,改!
一般在dtype=np.int64;dtype=np.float32中
(多数定义函数都有dtype属性)
最好int和float的位数要一致
import numpy as np
a = np.random.randint(100, size=(10**6), dtype="int64")
print(a)
print(type(a[0]))
更多推荐
已为社区贡献10条内容
所有评论(0)