g.edges[src, dst].data['label'] = label报错,信息:AttributeError: 'int' object has no attribute 'shape'
时间: 2023-08-22 15:14:20 浏览: 181
HighPass.m.zip_site:www.pudn.com
这个错误通常发生在使用DGL库时,它可能是因为你在定义图时,边的特征矩阵(例如label)的类型不正确,应该是一个数组或张量,而不是整数。你可以尝试将label转换为一个数组或张量,并将其分配给边的特征矩阵。例如,使用numpy数组:
```python
import numpy as np
g.edges[src, dst].data['label'] = np.array([label])
```
或使用torch张量:
```python
import torch
g.edges[src, dst].data['label'] = torch.tensor([label])
```
这将使你能够将label分配给边的特征矩阵,从而避免这个错误。
阅读全文