Pytorch将时序列表合并在一起。

Posted 2021-04-06

假设我有一个张量列表([A , B , C]，其中每个张量的形状是[batch_size X 1024]。 我想按以下方式将所有张量合并成一个单一的张量：A的第一行是新张量的第一行，B的第一行是新张量的第几行，C的第一行是新张量的第三行，以此类推。 到目前为止，我是用for循环来做的，但这完全没有效果。谢谢你

答案

这是一个最小的例子，它的工作原理。

import torch
a = torch.tensor([[1,1],[1,1]])
b = torch.tensor([[2,2],[2,2]])
c = torch.tensor([[3,3],[3,3]])
torch.stack([a,b,c],dim=0).view(6,2).t().contiguous().view(6,2)

输出是：

tensor([[1, 1],
        [2, 2],
        [3, 3],
        [1, 1],
        [2, 2],
        [3, 3]])

在你的例子中，view(6,2)应该改为 batch_size*3, 1024.

解决方案改编自 PyTorch论坛 其中以两个腾博会登录为例。