Fix the sample code of SELU/selu (#26565)

5 years ago · 87843bebde
parent 199b0c7c1b
commit 87843bebde
2 changed files with 2 additions and 2 deletions
--- a/python/paddle/nn/functional/activation.py
+++ b/python/paddle/nn/functional/activation.py
@ -665,7 +665,7 @@ def selu(x,
            paddle.disable_static()
-            x = paddle.to_tensor(np.array([[0, 1],[2, 3]]))
+            x = paddle.to_tensor(np.array([[0.0, 1.0],[2.0, 3.0]]))
            out = F.selu(x) # [[0, 1.050701],[2.101402, 3.152103]]
    """
    if in_dygraph_mode():
--- a/python/paddle/nn/layer/activation.py
+++ b/python/paddle/nn/layer/activation.py
@ -570,7 +570,7 @@ class SELU(layers.Layer):
            paddle.disable_static()
-            x = paddle.to_tensor(np.array([[0, 1],[2, 3]]))
+            x = paddle.to_tensor(np.array([[0.0, 1.0],[2.0, 3.0]]))
            m = paddle.nn.SELU()
            out = m(x) # [[0, 1.050701],[2.101402, 3.152103]]
    """