tensor zhuang list
时间: 2023-07-16 22:13:10 浏览: 72
pytorch改变tensor维度
将一个 Python 列表转换为 Tensor 可以使用 PyTorch 中的 torch.tensor() 函数。例如,以下代码将一个列表转换为 PyTorch 的 Tensor:
```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.tensor() 函数会自动进行类型转换。如果输入的列表为空,则会创建一个空的 Tensor。
阅读全文