如何交错 5 个 PyTorch 张量？

Question

我有 5 个形状张量torch.Size([7, 20, 180])

我想将它们一个接一个地交织在一起dim=1。这样我的最终形状就会是torch.Size([7, 100, 180])。

基本上，我想要第一个张量中的第一个元素，然后是第二个张量中的第一个元素，依此类推。

Answer 1

如果我理解正确的话

import torch
stacked = torch.stack(list_of_tensors, dim=2)
interleaved = torch.flatten(stacked, start_dim=1, end_dim=2)

interleaved显然是你需要的

（使用pytorch 1.1.0测试）