怎么将[(1,2,3,4)]的python列表转化为[[1,2,3,4]]的pytorch tensor
时间: 2024-01-01 13:05:57 浏览: 67
用python3读取python2的pickle数据方式
可以使用 PyTorch 中的 `torch.tensor()` 函数将 Python 列表转换为 PyTorch 张量。对于您提供的列表 `[(1,2,3,4)]`,您可以使用以下代码将其转换为形状为 (1, 4) 的 PyTorch 张量:
```
import torch
lst = [(1,2,3,4)]
tensor = torch.tensor(lst)
tensor = tensor.squeeze(0).unsqueeze(0)
print(tensor)
```
输出为:
```
tensor([[1, 2, 3, 4]])
```
在上述代码中,我们首先使用 `torch.tensor()` 函数将 Python 列表转换为 PyTorch 张量。由于原始列表中只有一个元素,因此得到的张量的形状为 (1, 4)。为了将其转换为形状为 (1, 1, 4) 的张量,我们使用 `squeeze()` 函数去除第一维的维度,然后使用 `unsqueeze()` 函数在第一维添加一个维度。
阅读全文