Add SpatialReSamplingEx (grouping all the Spatial*Sampling).

SpatialPadding can now pad on dimensions different than (2,3).
Add Tic/Toc modules, to time a network

1 files changed, 20 insertions, 0 deletions

diff --git a/Tic.lua b/Tic.lua
new file mode 100644
index 0000000..db50e55
--- /dev/null
+++ b/Tic.lua
@@ -0,0 +1,20 @@
+require 'nn'
+local Tic, parent = torch.class('nn.Tic', 'nn.Module')
+
+function Tic:__init(name)
+   parent.__init(self)
+   self.name = name or 'default'
+   tic_modules = tic_modules or {}
+   tic_modules[self.name] = torch.Timer()
+end
+
+function Tic:updateOutput(input)
+   tic_modules[self.name]:reset()
+   self.output = input
+   return self.output
+end
+
+function Tic:updateGradInput(input, gradOutput) 
+   self.gradInput = gradOutput
+   return self.gradInput
+end