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 81c34073b9f4f5e0b1f87f4623f86144f852d0f7..169404a5f767d09391afdff9a9c72bcbbb189324 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"