python将list变为tensor
时间: 2023-09-06 13:08:50 浏览: 176
python Tensor和Array对比分析
要将Python的list转换为PyTorch中的tensor,可以使用torch.Tensor()方法。例如,假设我们有一个包含3个元素的list,可以按照以下方式将其转换为tensor:
```python
import torch
my_list = [1, 2, 3]
my_tensor = torch.Tensor(my_list)
print(my_tensor)
```
输出为:
```
tensor([1., 2., 3.])
```
需要注意的是,这里生成的是浮点数类型的tensor。如果需要生成整数类型的tensor,可以使用`torch.LongTensor()`方法。
阅读全文