Machine learning 如何在Keras中反转形状以进行LSTM输入

Machine learning 如何在Keras中反转形状以进行LSTM输入,machine-learning,neural-network,tensorflow,deep-learning,keras,Machine Learning,Neural Network,Tensorflow,Deep Learning,Keras,我的输入状态为shape=(84,84,4) 这是连续帧的堆叠序列 我想把这个状态作为输入传递给keras模型, 先到时间分布层 然后-到LSTM 据我所知,时间步长是第一个维度 我需要适当地重塑我的状态以 shape=(4, 84, 84) 并以其自身的大小和拓扑保存帧 state_t=tf.transpose(state,[2,1,0]) 这就是你要找的吗? (或者[2,0,1],这取决于你想做什么…) state_t=tf.transpose(state,[2,1,0])

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

这是连续帧的堆叠序列

我想把这个状态作为输入传递给keras模型, 先到时间分布层 然后-到LSTM

据我所知,时间步长是第一个维度 我需要适当地重塑我的状态以

shape=(4, 84, 84)
并以其自身的大小和拓扑保存帧

state_t=tf.transpose(state,[2,1,0])
这就是你要找的吗?
(或者[2,0,1],这取决于你想做什么…)

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