将list转到tensor格式,python
时间: 2023-10-05 10:14:55 浏览: 82
tensor与numpy转换资源合集
5星 · 资源好评率100%
可以使用PyTorch或者TensorFlow等深度学习框架中的函数将list转换为tensor格式。
以PyTorch为例,可以使用torch.tensor()函数将list转换为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])
```
如果list中还包含多维数组,可以使用嵌套list的形式:
```python
my_list = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
my_tensor = torch.tensor(my_list)
print(my_tensor)
```
输出结果为:
```
tensor([[1, 2, 3],
[4, 5, 6],
[7, 8, 9]])
```
需要注意的是,如果list中的元素类型不一致,转换为tensor时会自动进行类型转换为最宽泛的类型。例如,如果list中既包含整数又包含浮点数,转换为tensor时会自动将整数转换为浮点数:
```python
my_list = [1, 2, 3, 4, 5.0]
my_tensor = torch.tensor(my_list)
print(my_tensor)
```
输出结果为:
```
tensor([1., 2., 3., 4., 5.])
```
阅读全文