问题和解决方案
-
运行GCN的例子的时候,出现了这个错误:
out = torch.from_numpy(out).to(torch.float) TypeError: expected np.ndarray (got matrix)
-
解决方案:
- 在
torch_geometric.io.planetoid.py
中添加import numpy as np
, - 将
out = torch.from_numpy(out).to(torch.float)
替换成:
out = torch.as_tensor(np.array(out).astype('float'))
- 在
-
搞定。(可能是版本的问题)