Implement a Recurrent Neural Net (RNN) in PyTorch! Learn how we can use the nn.RNN module and work with an input sequence. I also show you how easily we can switch to a gated recurrent unit (GRU) or long short-term memory (LSTM) RNN.
Code on GitHub: https://github.com/python-engineer/pytorch-examples.
- RNN: https://pytorch.org/docs/stable/generated/torch.nn.RNN.html
- LSTM: https://pytorch.org/docs/stable/generated/torch.nn.LSTM.html
- GRU: https://pytorch.org/docs/stable/generated/torch.nn.GRU.html