Python 展平三维张量

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

在展平之前，你需要排列你的轴，像这样：

t=t.swapdims（0,1）#（t，B，N）->（B，t，N）
t=t.view（B，-1）#（B，t，N）->（B，t*N）（相当于't.view（B，t*N）`）

谢谢！这似乎有效