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