diff --git a/python/paddle/v2/framework/tests/test_cross_entropy_op.py b/python/paddle/v2/framework/tests/test_cross_entropy_op.py index e693ba0ec280b8720630af29556fd645aae76e9b..fb6a440e23c26d1766bdf1fc5f24217afe1150f8 100644 --- a/python/paddle/v2/framework/tests/test_cross_entropy_op.py +++ b/python/paddle/v2/framework/tests/test_cross_entropy_op.py @@ -8,7 +8,8 @@ class TestCrossEntropy(OpTest): self.op_type = "onehot_cross_entropy" batch_size = 30 class_num = 10 - X = numpy.random.uniform(0.1, 1.0, [batch_size, class_num]).astype("float32") + X = numpy.random.uniform(0.1, 1.0, + [batch_size, class_num]).astype("float32") label = (class_num / 2) * numpy.ones(batch_size).astype("int32") self.inputs = {'X': X, 'label': label} Y = []