How to create a stacked-RNN model in Pytorch?

Question

I have a template of LSTM in Pytorch:

# RNN structure parameters
input_dim = 1
hidden_dim = 32
num_layers = 2
output_dim = 1
num_epochs = 50

class LSTM(nn.Mo