pytorch_geometric的Planetoid出现“TypeError: expected np.ndarray (got matrix)”的问题和解决方法

问题和解决方案

• 运行GCN的例子的时候，出现了这个错误：

  out = torch.from_numpy(out).to(torch.float)
  TypeError: expected np.ndarray (got matrix)
  

• 解决方案：

  1. 在torch_geometric.io.planetoid.py 中添加 import numpy as np,
  2. 将
     out = torch.from_numpy(out).to(torch.float)
     替换成:
     out = torch.as_tensor(np.array(out).astype('float'))

• 搞定。（可能是版本的问题）

参考

• pytorch论坛讨论