Welcome to mirror list, hosted at ThFree Co, Russian Federation.
1 2 3 4 5 6 7
local GRU, parent = torch.class('cudnn.GRU', 'cudnn.RNN') function GRU:__init(inputSize, hiddenSize, numLayers, batchFirst, dropout, rememberStates) parent.__init(self,inputSize, hiddenSize, numLayers, batchFirst, dropout, rememberStates) self.mode = 'CUDNN_GRU' self:reset() end