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