LogSoftMax.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.

LogSoftMax.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/LogSoftMax.lua

blob: fc63c4f18103c7f9a3b538f0fe9bbd5dc2076a47 (plain)

local SoftMax, parent = torch.class('cudnn.LogSoftMax', 'cudnn.SpatialSoftMax')

function SoftMax:updateOutput(input)
   self.mode = 'CUDNN_SOFTMAX_MODE_INSTANCE'
   self.algorithm = 'CUDNN_SOFTMAX_LOG'
   return parent.updateOutput(self, input)
end