Add w2nn.Print for debug

lib/Print.lua

@@ -0,0 +1,15 @@
+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

lib/PrintTable.lua

@@ -0,0 +1,16 @@
+local PrintTable, parent = torch.class('w2nn.PrintTable','nn.Module')
+
+function PrintTable:__init(id)
+   parent.__init(self)
+   self.id = id
+end
+function PrintTable:updateOutput(input)
+   print("----", self.id)
+   print(input)
+   self.output = input
+   return self.output
+end
+function PrintTable:updateGradInput(input, gradOutput)
+   self.gradInput = gradOutput
+   return self.gradInput
+end

lib/w2nn.lua

@@ -75,5 +75,7 @@ else
    require 'InplaceClip01'
    require 'L1Criterion'
    require 'ShakeShakeTable'
+   require 'PrintTable'
+   require 'Print'
    return w2nn
 end