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

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

blob: 3d86d0d04f3083729799fa7c35f1c2b388a95e50 (plain)

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

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