torch 如何将多个二维张量合并成一个三维张量，用for循环实现

要将多个二维张量合并成一个三维张量，可以使用torch.stack函数来实现。通过for循环可以实现如下代码：

import torch

# 生成3个2*2的二维张量
tensor_list = [torch.ones((2, 2)) * i for i in range(3)]

# 控制台打印出来看看
print(tensor_list)

# 使用for循环将多个二维张量合并成一个三维张量
stacked_tensor = None
for tensor in tensor_list:
    if stacked_tensor is None:
        stacked_tensor = tensor.unsqueeze(0)
    else:
        stacked_tensor = torch.cat((stacked_tensor, tensor.unsqueeze(0)), dim=0)

# 控制台打印出来看看
print(stacked_tensor)

其中`tensor_list`是需要合并的多个二维张量的列表。我们通过for循环遍历该列表，每次将一个二维张量通过`unsqueeze(0)`函数转化为一个三维张量，然后通过`torch.cat`函数将该三维张量与前面的所有三维张量进行拼接。最终得到的`stacked_tensor`就是我们需要的合并后的三维张量。注意，我们在以上代码中使用了`dim=0`参数来指定在哪个维度上进行拼接。