tensor转list
时间: 2023-10-05 20:06:42 浏览: 19
可以使用`tensor.tolist()`将tensor转为list。示例代码如下:
```python
import torch
# 创建一个tensor
tensor = torch.tensor([[1, 2], [3, 4]])
# 将tensor转为list
lst = tensor.tolist()
print(lst)
```
输出结果为:
```
[[1, 2], [3, 4]]
```
相关问题
tensor 转换list
将PyTorch中的list转换成tensor可以使用torch.Tensor()函数进行转换。例如:
```
import torch
a = [1, 2, 3]
tensor_a = torch.Tensor(a)
print(tensor_a)
```
输出结果为:
```
tensor([1., 2., 3.])
```
python tensor转list
可以使用 `tolist()` 方法将 Tensor 转换为 list。
示例代码:
```
import torch
# 创建一个 Tensor
tensor = torch.tensor([[1, 2], [3, 4]])
# 将 Tensor 转换为 list
lst = tensor.tolist()
print(lst)
```
输出结果:
```
[[1, 2], [3, 4]]
```
注意:如果 Tensor 中有浮点数,则转换后的 list 中也会是浮点数。如果需要将浮点数转换为整数,可以使用 `int()` 函数或者 `round()` 函数进行转换。