我的输入状态为shape =（84,84,4）
state = Input(shape=(84,84,4), dtype="float")
它是连续帧的堆叠序列 .
我想将此状态作为输入传递给keras模型，首先是 - 到TimeDistributed层然后 - 到LSTM
据我所知，时间步长是 first 维度，我需要 reshape 我的状态适当
shape=(4, 84, 84)
并保持框架的大小和拓扑
state_t=tf.transpose(state,[2,1,0])
这是你想要的 ？（或[2,0,1]取决于你想做什么......）
1 回答
这是你想要的 ？
（或[2,0,1]取决于你想做什么......）