From e519a1fc82e51f6a2f77e37f9c0ed99aa8b8ddca Mon Sep 17 00:00:00 2001 From: chengjuntao <18222160892@163.com> Date: Mon, 6 Jan 2020 10:02:27 +0800 Subject: [PATCH] add bigger shape test example for elementwise_pow OP (#22063) --- .../unittests/test_elementwise_pow_op.py | 20 +++++++++++++++++++ 1 file changed, 20 insertions(+) diff --git a/python/paddle/fluid/tests/unittests/test_elementwise_pow_op.py b/python/paddle/fluid/tests/unittests/test_elementwise_pow_op.py index 81c34073b9f..169404a5f76 100644 --- a/python/paddle/fluid/tests/unittests/test_elementwise_pow_op.py +++ b/python/paddle/fluid/tests/unittests/test_elementwise_pow_op.py @@ -35,6 +35,26 @@ class TestElementwisePowOp(OpTest): self.check_grad(['X', 'Y'], 'Out') +class TestElementwisePowOp_big_shape_1(TestElementwisePowOp): + def setUp(self): + self.op_type = "elementwise_pow" + self.inputs = { + 'X': np.random.uniform(0.1, 1, [100, 100]).astype("float64"), + 'Y': np.random.uniform(0.1, 1, [100, 100]).astype("float64") + } + self.outputs = {'Out': np.power(self.inputs['X'], self.inputs['Y'])} + + +class TestElementwisePowOp_big_shape_2(TestElementwisePowOp): + def setUp(self): + self.op_type = "elementwise_pow" + self.inputs = { + 'X': np.random.uniform(0.1, 1, [100, 100]).astype("float64"), + 'Y': np.random.uniform(0.1, 1, [100, 100]).astype("float64") * 20 + } + self.outputs = {'Out': np.power(self.inputs['X'], self.inputs['Y'])} + + class TestElementwisePowOp_scalar(TestElementwisePowOp): def setUp(self): self.op_type = "elementwise_pow" -- GitLab