python - 如何交错 5 个 PyTorch 张量？

Question

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

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

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

标签： pythonnumpypytorch

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 测试）