调用torch.Tensor类中的列表
时间: 2023-07-19 18:24:29 浏览: 99
Pytorch中torch.gather函数
5星 · 资源好评率100%
可以使用torch.Tensor类的构造函数将列表转换为张量。例如,以下代码将列表[1, 2, 3, 4]转换为张量:
```
import torch
my_list = [1, 2, 3, 4]
my_tensor = torch.Tensor(my_list)
print(my_tensor)
```
输出:
```
tensor([1., 2., 3., 4.])
```
注意,这将创建一个浮点型张量。如果你需要其他类型的张量,可以使用torch中其他的构造函数,例如`torch.LongTensor()`或`torch.FloatTensor()`等。
阅读全文