Merge pull request torch#273 from szagoruyko/master

moved zeroing to blas side in batch nn.Linear:forward
mys007 · May 23, 2015 · 1cca535 · 1cca535
2 parents 3dd5d1d + 8abe926
commit 1cca535
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Linear.lua b/Linear.lua
@@ -49,8 +49,8 @@ function Linear:updateOutput(input)
          self.output:copy(self.bias:view(1,nunit):expand(#self.output))
          self.output:select(2,1):addmv(1, input, self.weight:select(1,1))
       else
-         self.output:zero():addr(1, self.addBuffer, self.bias)
-         self.output:addmm(1, input, self.weight:t())
+         self.output:addmm(0, self.output, 1, input, self.weight:t())
+         self.output:addr(1, self.addBuffer, self.bias)
       end
    else
       error('input must be vector or matrix')