From 6e5eae137da7624ba5f0304b71c0cb38b8208f38 Mon Sep 17 00:00:00 2001 From: xuwei06 Date: Mon, 8 Jan 2018 17:05:38 -0800 Subject: [PATCH] Unittest for fluid.backward.calc_gradient() --- .../v2/fluid/tests/test_calc_gradient.py | 25 +++++++++++++++++++ 1 file changed, 25 insertions(+) create mode 100644 python/paddle/v2/fluid/tests/test_calc_gradient.py diff --git a/python/paddle/v2/fluid/tests/test_calc_gradient.py b/python/paddle/v2/fluid/tests/test_calc_gradient.py new file mode 100644 index 0000000000..c34c8ff6d1 --- /dev/null +++ b/python/paddle/v2/fluid/tests/test_calc_gradient.py @@ -0,0 +1,25 @@ +import unittest + +import paddle.v2.fluid as fluid +import paddle.v2.fluid.layers as layers +import paddle.v2.fluid.framework as framework +import paddle.v2.fluid.optimizer as optimizer +from paddle.v2.fluid.backward import calc_gradient + + +class TestCalcGradient(unittest.TestCase): + def test_calc_gradient(self): + x = layers.create_parameter(dtype="float32", shape=[5, 10]) + y = layers.create_parameter(dtype="float32", shape=[10, 8]) + mul_out = layers.mul(x=x, y=y) + mean_out = layers.mean(x=mul_out) + a = calc_gradient(mean_out, mul_out) + b = calc_gradient(mean_out, x) + place = fluid.CPUPlace() + exe = fluid.Executor(place) + exe.run(fluid.default_startup_program()) + exe.run(fluid.default_main_program(), feed={}, fetch_list=[a, b]) + + +if __name__ == "__main__": + unittest.main() -- GitLab