How to set number of LSTM blocks in a hidden layer?

1

Hi I am new to Pytorch.

In Keras,
model = Sequential()
model.add(LSTM(4, input_shape=(1, look_back)))
model.add(Dense(1))
model.compile(loss=‘mean_squared_error’, optimizer=‘adam’)
model.fit(trainX, trainY, epochs=100, batch_size=1, verbose=2)

4 means a hidden layer with 4 LSTM blocks.

How to set this parameter in Pytorch?
Is it the hidden size? (“hidden_size – The number of features in the hidden state h”)

Thanks.

2

Yup it’s the hidden size

3

Thank you very much.
I understand more about LSTM.