cudnn.torch/SpatialLogSoftMax.lua at master · soumith/cudnn.torch · GitHub

Sitelet https://github.com/soumith/cudnn.torch/blob/master/SpatialLogSoftMax.lua

7 lines (6 loc) · 235 Bytes

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'