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

Question

我的输入状态为shape = (84,84,4)

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

它是连续帧的堆叠序列。

我想把这个状态作为输入传递给keras模型，首先-到time分布层然后- to LSTM

据我所知，时间步长是第一个维度我需要将的状态调整为

shape=(4, 84, 84)

Answer 1

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

这是你要找的吗?
(或[2,0,1]，这取决于你想做什么…)