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

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

blob: ac222e62ae2450b593bc98b04cc63e8c8d09487a (plain)

local Tanh, parent = torch.class('cudnn.Tanh','cudnn._Pointwise')

function Tanh:__init(inplace)
   parent.__init(self, inplace)
   self.mode = 'CUDNN_ACTIVATION_TANH'
end

function Tanh:updateOutput(input)
  if not self.mode then self.mode = 'CUDNN_ACTIVATION_TANH' end
  return parent.updateOutput(self, input)
end