testing creating a custom rnn layer with pytorch - View it on GitHub