torch 张量转列表
时间: 2023-11-13 16:06:20 浏览: 32
可以使用 `.tolist()` 方法将张量转换为列表。例如:
```
import torch
# 创建一个张量
t = torch.tensor([[1, 2], [3, 4]])
# 将张量转换为列表
lst = t.tolist()
print(lst) # [[1, 2], [3, 4]]
```
相关问题
列表转换为torch张量
可以使用PyTorch中的torch.tensor()方法将列表转换为torch张量。示例代码如下:
```python
import torch
my_list = [1, 2, 3, 4, 5]
my_tensor = torch.tensor(my_list)
print(my_tensor)
```
输出结果为:
```
tensor([1, 2, 3, 4, 5])
```
torch numpy转张量
可以使用`torch.from_numpy()`函数将numpy数组转换为PyTorch张量。
示例代码:
```
import numpy as np
import torch
# 创建一个numpy数组
arr = np.array([[1, 2, 3], [4, 5, 6]])
# 将numpy数组转换为PyTorch张量
tensor = torch.from_numpy(arr)
print(tensor)
```
输出:
```
tensor([[1, 2, 3],
[4, 5, 6]], dtype=torch.int32)
```