Merge pull request torch#258 from adampolyak/master

Buffers for PReLU cuda implementation.
mys007 · May 5, 2015 · 5d3aaa0 · 5d3aaa0
2 parents 1260c83 + e39bfeb
commit 5d3aaa0
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/PReLU.lua b/PReLU.lua
@@ -6,6 +6,8 @@ function PReLU:__init(nOutputPlane)
    self.nOutputPlane = nOutputPlane or 0
    self.weight = torch.Tensor(nOutputPlane or 1):fill(0.25)
    self.gradWeight = torch.Tensor(nOutputPlane or 1)
+   self.gradWeightBuf = torch.Tensor()
+   self.gradWeightBuf2 = torch.Tensor()
 end
 
 function PReLU:updateOutput(input)