nn/Power.lua at bmmm · dominikgrewe/nn

Repository navigation

local Power, parent = torch.class('nn.Power','nn.Module')

function Power:__init(p)

parent.__init(self)

self.pow = p

if not p then

error('nn.Power(power)')

end

function Power:updateOutput(input)

self.output:resizeAs(input):copy(input)

self.output:pow(self.pow)

return self.output

end

function Power:updateGradInput(input, gradOutput)

self.gradInput:resizeAs(input):copy(gradOutput)

self.gradInput:cmul(self.output):cdiv(input):mul(self.pow)

return self.gradInput

end