from keras.models import Sequential from keras.layers import Dense from keras.layers import LSTM net = Sequential() net.add(LSTM(units = 3, return_sequences = True, input_shape = (X_train.shape, 1))) net.add(LSTM(units = 3)) net.add(Dense(units = 1))
Below image is an attempt to visually represent the above code snippet.
From the above figure, I understand that there are two layers of LSTM units. Each of those layers has three LSTM units/neurons. There is a single output neuron/unit. The same input is fed to all three LSTM units in the first layer.
a) Does the figure interpret the code sample correctly?
b) Have I described/understood the figure correct?