keras.layers.simplernn
inputs = np.random.random([32, 10, 8]).astype(np.float32)simple_rnn = tf.keras.layers.SimpleRNN(4)output = simple_rnn(inputs) # The output has shape `[32, 4]`.simple_rnn = tf.keras.layers.SimpleRNN( 4, return_sequences=True, return_state=True)# whole_sequence_output has shape `[32, 10, 4]`.# final_state has shape `[32, 4]`.whole_sequence_output, final_state = simple_rnn(inputs)
Worried Wallaby