torch / rnn

Torch recurrent neural networks
BSD 3-Clause "New" or "Revised" License
64 stars 17 forks source link

C implementation of LSTM #31

Open nkoumchatzky opened 7 years ago

nkoumchatzky commented 7 years ago

Pure C implementation for LSTMs with basic LSTM cell, including pre-multiplication and CUDA-like input format [(sum(T_i)) X outputSize], where T_i are the number of timesteps for each element of the batch, in decreasing value.