16 lines
372 B
Lua
16 lines
372 B
Lua
local Print, parent = torch.class('w2nn.Print','nn.Module')
|
|
|
|
function Print:__init()
|
|
parent.__init(self)
|
|
end
|
|
function Print:updateOutput(input)
|
|
print(input:size())
|
|
self.output:resizeAs(input)
|
|
self.output:copy(input)
|
|
return self.output
|
|
end
|
|
function Print:updateGradInput(input, gradOutput)
|
|
self.gradInput:resizeAs(GradOutput)
|
|
return self.gradInput
|
|
end
|