indices = list(itertools.combinations(range(features.shape[0]), 2))报错AttributeError: 'list' object has no attribute 'shape'
时间: 2023-09-20 13:12:10 浏览: 121
python报错: 'list' object has no attribute 'shape'的解决
5星 · 资源好评率100%
抱歉,我之前的代码有误。`features`应该是一个numpy数组或者PyTorch张量,而不是一个普通的Python列表才能使用`shape`属性。请将代码中的`features`替换为以下代码,即可解决该问题:
```python
import numpy as np
features = np.array([
[0.1, 0.2, 0.3],
[0.2, 0.4, 0.1],
[0.3, 0.1, 0.5],
[0.4, 0.3, 0.2]
], dtype=np.float32)
```
或者使用PyTorch张量:
```python
import torch
features = torch.tensor([
[0.1, 0.2, 0.3],
[0.2, 0.4, 0.1],
[0.3, 0.1, 0.5],
[0.4, 0.3, 0.2]
], dtype=torch.float)
```
这样就可以使用`features.shape[0]`获取节点数量了。
阅读全文