diff options
author | SeanNaren <taz838@hotmail.co.uk> | 2016-04-11 22:21:40 +0300 |
---|---|---|
committer | Boris Fomitchev <bfomitchev@nvidia.com> | 2016-04-18 23:19:26 +0300 |
commit | 96f13324d07b0b80fb429f6dbe35fa5402234968 (patch) | |
tree | 1440a87a19522ef62d5b28ba98e327591c4796f0 /LSTM.lua | |
parent | 60a66872730eceaf4f769c281e2ad7289272323e (diff) |
Added tests, modified README and added RNN modules
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 |