diff options
Diffstat (limited to 'LSTM.lua')
| -rw-r--r-- | LSTM.lua | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/LSTM.lua b/LSTM.lua new file mode 100644 index 0000000..29c199c --- /dev/null +++ b/LSTM.lua @@ -0,0 +1,7 @@ +local LSTM, parent = torch.class('cudnn.LSTM', 'cudnn.RNN') + +function LSTM:__init(inputSize, hiddenSize, numLayers, batchFirst) + parent.__init(self,inputSize, hiddenSize, numLayers, batchFirst) + self.mode = 'CUDNN_LSTM' + self:reset() +end |