Welcome to mirror list, hosted at ThFree Co, Russian Federation.
1 2 3 4 5 6
local Sigmoid, parent = torch.class('cudnn.Sigmoid','cudnn._Pointwise') function Sigmoid:updateOutput(input) if not self.mode then self.mode = 'CUDNN_ACTIVATION_SIGMOID' end return parent.updateOutput(self, input) end