Welcome to mirror list, hosted at ThFree Co, Russian Federation.

Square.lua - github.com/torch/nn.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
blob: c1b80dca5a74484b8db6bb7ef0de2edab28eca07 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
local Square, parent = torch.class('nn.Square','nn.Module')

function Square:__init(args)
   parent.__init(self)
end

function Square:updateOutput(input)
   return input.nn.Square_updateOutput(self, input)
end

function Square:updateGradInput(input, gradOutput)
   return input.nn.Square_updateGradInput(self, input, gradOutput)
end