pt文件转onnx AttributeError: 'list' object has no attribute 'shape'
时间: 2023-12-27 20:05:04 浏览: 126
python报错: 'list' object has no attribute 'shape'的解决
5星 · 资源好评率100%
在进行pt文件转onnx的过程中,出现了错误提示"AttributeError: 'list' object has no attribute 'shape'"。这个错误通常是由于将一个列表对象传递给了需要具有shape属性的函数或方法而导致的。
为了解决这个问题,你可以确保你传递给转换函数的是一个具有shape属性的对象,例如numpy数组或torch张量。你可以使用以下方法来解决这个问题:
1. 将列表转换为numpy数组:
```python
import numpy as np
# 假设你的列表是list_data
array_data = np.array(list_data)
```
2. 将列表转换为torch张量:
```python
import torch
# 假设你的列表是list_data
tensor_data = torch.tensor(list_data)
```
通过将列表转换为numpy数组或torch张量,你可以确保传递给转换函数的对象具有shape属性,从而避免出现"AttributeError: 'list' object has no attribute 'shape'"的错误。
阅读全文