JustTable.lua


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


local JustTable, parent = torch.class('nngraph.JustTable', 'nn.Module')
function JustTable:__init()
   self.output = {}
end

-- The input is one element.
-- The output is a table with one element: {element}
function JustTable:updateOutput(input)
   self.output[1] = input
   return self.output
end

function JustTable:updateGradInput(input, gradOutput)
   self.gradInput = gradOutput[1]
   return self.gradInput
end