Assert that dim is integer in Normalize

torch · Apr 13, 2016 · bae3b67 · bae3b67
1 parent ded85de
commit bae3b67
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/Normalize.lua b/Normalize.lua
@@ -12,6 +12,7 @@ end
 function Normalize:updateOutput(input)
   assert(math.abs(self.dim) <= input:dim(), 
     'input has less dimensions than the normalization dimension')
+  assert(self.dim % 1 == 0, 'dimension should be an integer')
   local dim = self.dim
   if dim < 0 then
     dim = input:dim() + dim + 1
@@ -47,6 +48,7 @@ end
 function Normalize:updateGradInput(input, gradOutput)
   assert(math.abs(self.dim) <= input:dim(), 
     'input has less dimensions than the normalization dimension')
+  assert(self.dim % 1 == 0, 'dimension should be an integer')
   local dim = self.dim
   if dim < 0 then
     dim = input:dim() + dim + 1