We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ded85de commit bae3b67Copy full SHA for bae3b67
Normalize.lua
@@ -12,6 +12,7 @@ end
12
function Normalize:updateOutput(input)
13
assert(math.abs(self.dim) <= input:dim(),
14
'input has less dimensions than the normalization dimension')
15
+ assert(self.dim % 1 == 0, 'dimension should be an integer')
16
local dim = self.dim
17
if dim < 0 then
18
dim = input:dim() + dim + 1
@@ -47,6 +48,7 @@ end
47
48
function Normalize:updateGradInput(input, gradOutput)
49
50
51
52
53
54
0 commit comments