/torch_rnn

Implementation of pytorch sequence models from scratch but allows defining output shape as parameter

Primary LanguagePython