for medium article on LSTMs

lstm01.py

input_size = 1    # The number of variables in your sequence data. 
n_hidden   = 100  # The number of hidden nodes in the LSTM layer.
n_layers   = 2    # The total number of LSTM models layers
out_size   = 1    # The size of the output you desire from your RNN

lstm   = nn.LSTM(input_size, n_hidden, n_layers)
linear = nn.Linear(n_hidden, 1)