ValueError: Expected 'edge_index' to be of integer type (got 'torch.float32')
时间: 2023-08-08 16:13:22 浏览: 520
python3.7解决最小二乘遇到ValueError:Expected 2D array, got 1D array instead: array=[5.].关于reshape和predict
这个错误通常意味着您的 PyTorch 张量中包含了错误的数据类型。在大多数情况下,Graph Neural Network 中的边索引应该是整数类型,因为它们代表节点之间的连接。
您可以通过将边索引张量转换为整数类型来解决此错误。例如,如果您的边索引张量是 `edge_index`,则可以使用以下代码将其转换为整数类型:
```
edge_index = edge_index.to(torch.int64)
```
如果您在处理其他类型的张量时遇到类似的错误,也可以使用类似的方法将其转换为正确的数据类型。
阅读全文