search

google / objax

Apache License 2.0
769 stars 77 forks source link

Add lstm cell #188

Closed mariosasko closed 2 years ago

mariosasko commented 3 years ago

This PR adds LSTM cell.

To test the output, this cell's output is compared with the output of PyTorch's LSTM cell.

Closes #69