nn.Mean works with CudaTensors

1 files changed, 1 insertions, 1 deletions

diff --git a/Mean.lua b/Mean.lua
index 55e7609..ebd69bd 100644
--- a/Mean.lua
+++ b/Mean.lua
@@ -7,7 +7,7 @@ function Mean:__init(dimension)
 end
 
 function Mean:updateOutput(input)
-   input.torch.mean(self.output, input, self.dimension)
+   self.output:mean(input, self.dimension)
    self.output = self.output:select(self.dimension, 1)
    return self.output
 end