如何将torch.Tensor类型转换为list
时间: 2024-05-11 12:19:18 浏览: 148
pytorch numpy list类型之间的相互转换实例
5星 · 资源好评率100%
可以使用`tolist()`方法将`torch.Tensor`类型转换为Python中的`list`类型。例如:
```python
import torch
# 创建一个tensor
tensor = torch.tensor([1, 2, 3])
# 将tensor转换为list
lst = tensor.tolist()
print(lst) # 输出 [1, 2, 3]
```
注意:如果`tensor`是多维的,则`tolist()`方法会递归地将其所有元素转换为`list`类型。
阅读全文