Faster reset for 'nn' modules.

1 files changed, 1 insertions, 6 deletions

diff --git a/Euclidean.lua b/Euclidean.lua
index 808b7ab..ca0dc33 100644
--- a/Euclidean.lua
+++ b/Euclidean.lua
@@ -20,12 +20,7 @@ function Euclidean:reset(stdv)
    else
       stdv = 1./math.sqrt(self.weight:size(1))
    end
-
-   for i=1,self.weight:size(2) do
-      self.weight:select(2, i):apply(function()
-                                        return torch.uniform(-stdv, stdv)
-                                     end)
-   end
+   self.weight:uniform(-stdv, stdv)
 end
 
 function Euclidean:updateOutput(input)