list转tensor
时间: 2023-04-11 09:00:39 浏览: 223
tensor与numpy转换资源合集
5星 · 资源好评率100%
可以使用 PyTorch 中的 torch.tensor() 方法将 list 转换为 tensor。例如,如果有一个包含 3 个元素的 list,可以使用以下代码将其转换为 tensor:
```
import torch
my_list = [1, 2, 3]
my_tensor = torch.tensor(my_list)
```
这将创建一个形状为 (3,) 的 tensor,其中包含了 list 中的元素。
阅读全文