使用了_Qs = _Qs.type(torch.int64)还是出现那个问题
时间: 2024-02-26 15:52:42 浏览: 71
torch_scatter-2.1.2-cp310-cp310-macosx_11_0_x86_64.whl.zip
在使用`gather`函数时,如果出现`RuntimeError: gather(): Expected dtype int64 for index`错误,可能是因为你正在使用的是非int64类型的张量进行索引,或者索引张量中包含了非int64类型的值。以下是一些可能的解决方法:
1. 确保你的索引张量是int64类型。可以使用`.type(torch.int64)`方法将整个张量转换为int64类型。
2. 确保你的索引张量中不包含非int64类型的值。可以使用`.type(torch.int64)`方法将整个张量转换为int64类型,然后检查索引张量中是否有非int64类型的值。
3. 确保你使用的是int64类型的张量进行索引。可以使用`.type(torch.int64)`方法将整个张量转换为int64类型,然后再进行索引操作。
如果你已经尝试了以上解决方法,但仍然遇到了问题,请提供更多的代码和错误信息,以便我更好地帮助你解决问题。
阅读全文