ClippedReLU.lua


1
2
3
4
5
6
7
8
9
10
11
12
13
14

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

ClippedReLU.mode = 'CUDNN_ACTIVATION_CLIPPED_RELU'

function ClippedReLU:__init(ceiling, inplace)
    parent.__init(self)
    assert(ceiling, "No ceiling was given to ClippedReLU")
    self.ceiling = ceiling
    self.inplace = inplace or false
end

function ClippedReLU:updateOutput(input)
    return parent.updateOutput(self, input)
end