Can you write me a complete LSTM neural network from scratch while using the PyTorch module?