18 lines
416 B
Lua
18 lines
416 B
Lua
local Print, parent = torch.class('w2nn.Print','nn.Module')
|
|
|
|
function Print:__init(id)
|
|
parent.__init(self)
|
|
self.id = id
|
|
end
|
|
function Print:updateOutput(input)
|
|
print("----", self.id)
|
|
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
|