BGRU.lua - github.com/soumith/cudnn.torch.git - Unnamed repository; edit this file 'description' to name the repository.

Welcome to mirror list, hosted at ThFree Co, Russian Federation.

BGRU.lua - github.com/soumith/cudnn.torch.git - Unnamed repository; edit this file 'description' to name the repository.

summary refs log tree commit diff

path: root/BGRU.lua

blob: 31d5d923382f8cc59ba76043bd2917a76084d84c (plain)

local BGRU, parent = torch.class('cudnn.BGRU', 'cudnn.RNN')

function BGRU:__init(inputSize, hiddenSize, numLayers, batchFirst, dropout)
    parent.__init(self, inputSize, hiddenSize, numLayers, batchFirst, dropout)
    self.bidirectional = 'CUDNN_BIDIRECTIONAL'
    self.mode = 'CUDNN_GRU'
    self.numDirections = 2
    self:reset()
end