graph.ndata["node_features"] = new_node_feat报错AttributeError: 'numpy.ndarray' object has no attribute 'device'
时间: 2023-12-15 13:05:39 浏览: 165
AttributeError: module 'tensorflow.compat.v1' has no attribute '
这个错误提示也是 PyTorch 的错误,可能是因为你将 numpy.ndarray 转换为 PyTorch 的 Tensor 时没有指定使用 CPU 进行操作,而在之后使用时调用了 GPU 相关的函数,导致出现错误。
你可以将 numpy.ndarray 转换为 PyTorch 的 Tensor 时指定使用 CPU 进行操作,可以使用如下代码实现:
```
import torch
# 将 numpy.ndarray 转换为 PyTorch 的 Tensor,并指定使用 CPU 进行操作
tensor = torch.from_numpy(numpy_array).cpu()
# 将 Tensor 赋值给图的节点特征
graph.ndata["node_features"] = tensor
# 进行相应的操作
```
这样就可以避免 'numpy.ndarray' object has no attribute 'device' 错误了。
阅读全文