将不同长度的 2D 张量列表转换为一个 3D 张量

Question

我有一个包含 3 个张量的列表，其形状为： (8, 2), (8, 4), (8, 6)

我想将此列表转换为以下形状： ( 8, 3, x)

我该怎么做？我知道我需要使用 torch.cat、torch.stack 和 torch.transpose 的某种组合，但我无法弄清楚。

提前致谢！

Answer 1

正如您所说，您需要使用torch.cat，还需要torch.reshape。假设如下：

a = torch.rand(8,2)
b = torch.rand(8,4)
c = torch.rand(8,6)

并假设确实可以将张量重塑为 (8,3,-1) 形状，其中 -1 代表只要需要的话：

d = torch.cat((a,b,c), dim=1)
e = torch.reshape(d, (8,3,-1))

我会解释一下。由于 a,b,c 中的第一维不同，因此串联必须沿第一维进行，如变量 d 中所示。然后，您可以重塑张量，如 e 中所示，其中 -1 代表“只要需要”。