我的输入状态形状= (84,84,4)
state = Input(shape=(84,84,4), dtype="float")
它是连续帧的堆叠序列。
我想将此状态传递给 keras 模型作为输入, 首先 - TimeDistributed 层 然后 - LSTM
据我了解,时间步长是第一个维度 我需要适本地 reshape 我的状态
shape=(4, 84, 84)
并将框架保持在自己的尺寸和拓扑中
最佳答案
state_t=tf.transpose(state,[2,1,0])
这是您要找的吗?
(或 [2,0,1] 这取决于你想做什么......)
关于machine-learning - 如何在 Keras 中反转 LSTM 输入的形状,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/40155970/