将二维列表转化为三维张量
时间: 2023-08-15 08:05:35 浏览: 187
unfold_tall3.zip_tensor unfold_三阶张量_张量_张量展开矩阵
要将二维列表转化为三维张量,您可以使用`torch.Tensor()`函数,并指定所需的维度。假设您的二维列表`my_list`是一个m行n列的矩阵,可以按照以下步骤进行转换:
```python
import torch
my_list = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
my_tensor = torch.Tensor(my_list)
# 添加一个维度,将二维列表转化为三维张量
my_tensor = my_tensor.unsqueeze(0)
print(my_tensor)
```
在上述示例中,`my_list`是一个二维列表,其中包含3个子列表。通过将`my_list`传递给`torch.Tensor()`函数,您将得到一个二维张量`my_tensor`。然后,通过使用`unsqueeze()`函数添加一个维度,将二维张量转化为三维张量。运行代码后,您将看到输出的张量对象`my_tensor`,其值与输入的二维列表相同。请注意,添加的维度将放置在最前面,所以在这个例子中,张量的形状将变为(1, 3, 3)。
阅读全文