Add mean op unit test in python

paddle/operators/mean_op.cu

@@ -1,4 +1,5 @@
-#include "paddle/framework/op_registry.h"
+#define EIGEN_USE_GPU
+
 #include "paddle/operators/mean_op.h"
 
-REGISTER_OP_GPU_KERNEL(mean, ops::AddKernel<ops::GPUPlace, float>);
+REGISTER_OP_GPU_KERNEL(mean, ops::MeanKernel<ops::GPUPlace, float>);

paddle/operators/mean_op.h

@@ -26,8 +26,8 @@ public:
     auto output = context.Output(0)->GetMutable<Tensor>();
 
     output->mutable_data<T>(context.GetPlace());
-    EigenVector<T>::Flatten(*output).device(
+
-        *(context.GetEigenDevice<Place>())) =
+    EigenScalar<T>::From(*output).device(*(context.GetEigenDevice<Place>())) =
         EigenVector<T>::Flatten(input).mean();
   }
 };

paddle/pybind/CMakeLists.txt

@@ -1,2 +1,9 @@
-cc_library(paddle_pybind SHARED SRCS pybind.cc DEPS pybind python
+cc_library(paddle_pybind SHARED
-        add_op fc_op sgd_op cross_entropy_op recurrent_network_op)
+    SRCS pybind.cc
+    DEPS pybind python
+	fc_op
+	sgd_op
+	add_op
+	mean_op
+	cross_entropy_op
+	recurrent_network_op)

paddle/pybind/pybind.cc

@@ -33,6 +33,7 @@ USE_OP(onehot_cross_entropy);
 USE_OP_WITHOUT_KERNEL(fc);
 USE_OP(sgd);
 USE_OP(mul);
+USE_OP(mean);
 USE_OP(sigmoid);
 USE_OP(softmax);
 USE_OP(rowwise_add);

python/paddle/v2/framework/tests/CMakeLists.txt

@@ -10,6 +10,7 @@ add_python_test(test_framework
     test_sgd_op.py
     test_cross_entropy_op.py
     test_mul_op.py
+    test_mean_op.py
     test_sigmoid_op.py
     test_softmax_op.py
     test_rowwise_add_op.py

python/paddle/v2/framework/tests/test_mean_op.py

@@ -0,0 +1,16 @@
+import unittest
+from op_test_util import OpTestMeta
+import numpy as np
+
+
+class TestMeanOp(unittest.TestCase):
+    __metaclass__ = OpTestMeta
+
+    def setUp(self):
+        self.type = "mean"
+        self.X = np.random.random((32, 784)).astype("float32")
+        self.Out = np.mean(self.X)
+
+
+if __name__ == '__main__':
+    unittest.main()