VolumetricLogSoftMax.lua - github.com/soumith/cudnn.torch.git - Unnamed repository; edit this file 'description' to name the repository.

Welcome to mirror list, hosted at ThFree Co, Russian Federation.

VolumetricLogSoftMax.lua - github.com/soumith/cudnn.torch.git - Unnamed repository; edit this file 'description' to name the repository.

summary refs log tree commit diff

path: root/VolumetricLogSoftMax.lua

blob: a23ed602cb9470e49a6259cd8ed8dc33da16fd3e (plain)

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'
end