diff --git a/python/paddle/nn/functional/norm.py b/python/paddle/nn/functional/norm.py index 4a6d4abfe86b605d8fbf0138630eb220a566de7f..0beedd96eb70e52706cf8d2b58ffaf0e611a2855 100644 --- a/python/paddle/nn/functional/norm.py +++ b/python/paddle/nn/functional/norm.py @@ -110,7 +110,8 @@ def normalize(x, p=2, axis=1, epsilon=1e-12, name=None): type='p_norm', inputs={'X': x}, outputs={'Out': out}, attrs=attrs) eps = out.block.create_var(dtype=out.dtype) paddle.fluid.layers.fill_constant([1], out.dtype, epsilon, out=eps) - return paddle.fluid.layers.elementwise_div(x, paddle.maximum(out, eps), name=name) + return paddle.fluid.layers.elementwise_div( + x, paddle.maximum(out, eps), name=name) def batch_norm(x,