refine leaky_relu docs (#26558)

5 years ago · 81c127d4ec
parent 66596bd20f
commit 81c127d4ec
2 changed files with 7 additions and 7 deletions
--- a/python/paddle/nn/functional/activation.py
+++ b/python/paddle/nn/functional/activation.py
@ -419,7 +419,7 @@ def leaky_relu(x, negative_slope=0.01, name=None):

            paddle.disable_static()

-            x = paddle.to_tensor(np.array([-2, 0, 1]))
+            x = paddle.to_tensor(np.array([-2, 0, 1], 'float32'))
            out = F.leaky_relu(x) # [-0.02, 0., 1.]

    """
--- a/python/paddle/nn/layer/activation.py
+++ b/python/paddle/nn/layer/activation.py
@ -625,7 +625,7 @@ class LeakyReLU(layers.Layer):
            paddle.disable_static()

            m = paddle.nn.LeakyReLU()
-            x = paddle.to_tensor(np.array([-2, 0, 1]))
+            x = paddle.to_tensor(np.array([-2, 0, 1], 'float32'))
            out = m(x)  # [-0.02, 0., 1.]
    """