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

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

7 lines (6 loc) · 249 Bytes

local SoftMax, parent = torch.class('cudnn.VolumetricLogSoftMax', 'cudnn.VolumetricSoftMax')
function SoftMax:__init(fast)
   parent.__init(self, fast)
   self.ssm.mode = 'CUDNN_SOFTMAX_MODE_CHANNEL'
   self.ssm.algorithm = 'CUDNN_SOFTMAX_LOG'