Pytorch, Pytorch Rnn, Pytorch Mnist Dataset, Build Model With Pytorch.
of the input at each time step Hidden dimension — represents the size of the hidden state and cell state at each time step Number of layers — the number of LSTM layers stacked on top of each other Num of classes — require an output layer with 10 nodes in order to predict the probability