将tensor转化为list
时间: 2023-06-11 12:10:20 浏览: 91
pytorch中tensor张量数据类型的转化方式
5星 · 资源好评率100%
可以使用 PyTorch 中的 `tolist()` 方法将 tensor 转换为 list,示例如下:
```python
import torch
tensor = torch.tensor([1, 2, 3])
lst = tensor.tolist()
print(lst) # [1, 2, 3]
```
注意,如果 tensor 中包含多维数据,则需要递归使用 `tolist()` 方法将其转换为嵌套的 list。
阅读全文