如何交错 5 个 PyTorch 张量？答案

【问题标题】：How can I interleave 5 PyTorch tensors?如何交错 5 个 PyTorch 张量？
【发布时间】：2020-07-07 05:17:27
【问题描述】：

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

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

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

【问题讨论】：

【解决方案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 测试）

【讨论】：