使用summary时报错'list' object has no attribute 'size'怎么办
时间: 2023-09-21 08:11:12 浏览: 125
AttributeError: ‘NoneType’ object has no attribute ‘children’ 错误
5星 · 资源好评率100%
这个错误通常是因为你在处理列表对象时使用了PyTorch的size()函数,而size()函数只能用于PyTorch的tensor对象。你需要将列表对象转换为tensor对象才能使用size()函数。
例如,如果你要将一个列表对象转换为tensor对象,可以使用torch.tensor()函数,如下所示:
```
import torch
my_list = [1, 2, 3, 4, 5]
my_tensor = torch.tensor(my_list)
print(my_tensor.size())
```
在上面的代码中,我们首先创建了一个列表对象my_list,然后使用torch.tensor()函数将其转换为tensor对象my_tensor,并使用size()函数获取其大小。
如果你要在使用summary时转换列表对象为tensor对象,请在代码中寻找转换列表对象为tensor对象的代码,并确保使用了torch.tensor()函数。
阅读全文