python - 如何交错放置 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 测试)