在张量元组列表中堆叠张量

Question

我有一个 PyTorch 张量元组的列表。它看起来像这样：

[
    (tensor([1, 2, 3]),  tensor([4, 5, 6, 7]),  tensor([8])),
    (tensor([9, 10,11]), tensor([11,12,13,14]), tensor([15])),
    (tensor([16,17,18]), tensor([19,20,21,22]), tensor([23])),
    ...
]

每列中的张量（即，位于其各自元组的 k 位置的张量）共享相同的形状。我想在每列中堆叠张量，以便最终得到一个元组，每个值都是沿列维度连接的张量。

在这种情况下，输出元组将具有三个值，如下所示：

(
 tensor([[1,2,3], [9,10,11], [16,17,18]]),

 tensor([[4,5,6,7], [11,12,13,14], [19,20,21,22]],

 tensor([[8],[15],[23])
)

这是一个虚构的例子。我想对任意长度的元组和任意大小的张量执行此操作。使用 PyTorch 快速执行此类连接的最佳方法是什么？

Answer 1

如果有人让自己陷入同样复杂的场景，我可以用一个可爱的单线来解决它：

tuple(map(torch.stack, zip(*x)))

在这种情况下，x 是我上面提到的原始列表。这行代码将x 转换为所需的确切格式。