Welcome to mirror list, hosted at ThFree Co, Russian Federation.
1 2 3 4 5 6 7
local SoftMax, parent = torch.class('cudnn.LogSoftMax', 'cudnn.SpatialSoftMax') function SoftMax:updateOutput(input) self.mode = 'CUDNN_SOFTMAX_MODE_INSTANCE' self.algorithm = 'CUDNN_SOFTMAX_LOG' return parent.updateOutput(self, input) end