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