展平 3D 张量

【问题标题】：Flatten 3D tensor展平 3D 张量
【发布时间】：2021-08-08 05:22:32
【问题描述】：

我有一个形状为 T x B x N 的张量（RNN 的训练数据，T 是最大 seq 长度，B 是批次数，N 个特征），我想展平所有特征跨时间步长，这样我就得到了一个形状为 B x TN 的张量。还没有弄清楚如何做到这一点..

【问题讨论】：

【解决方案1】：

您需要在展平之前排列轴，如下所示：

t = t.swapdims(0,1) # (T,B,N) -> (B,T,N)
t = t.view(B,-1)    # (B,T,N) -> (B,T*N) (equivalent to `t.view(B,T*N)`)

【讨论】：