如何在Keras中反转LSTM输入的形状

Question

我的输入状态为shape =（84,84,4）

tests
---dir1
   test1.py
-------sub-dir1-1
       test-1-1.py
-------sub-dir1-2
       test-1-2.py
---dir2
   test2.py
-------sub-dir2-1
       test-2-1.py
-------sub-dir2-2
       test-2-2.py

它是连续帧的堆叠序列。

我想将此状态作为输入传递给keras模型， 首先 - 到TimeDistributed层 然后 - 到LSTM

据我所知，时间步长是第一个维度 我需要重塑我的状态

state = Input(shape=(84,84,4), dtype="float")

并保持帧的大小和拓扑

Answer 1

state_t=tf.transpose(state,[2,1,0])

这是你要找的吗？
（或[2,0,1]取决于你想做什么......）