From 933868cd2c7d326b6980dc4ce97bea2711d86f4b Mon Sep 17 00:00:00 2001
From: limingshu <61349199+JamesLim-sy@users.noreply.github.com>
Date: Fri, 19 Aug 2022 10:44:22 +0800
Subject: [PATCH] first commit (#45254)

---
 python/paddle/nn/layer/transformer.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/python/paddle/nn/layer/transformer.py b/python/paddle/nn/layer/transformer.py
index 35acaded2ea..75f8bac75bc 100644
--- a/python/paddle/nn/layer/transformer.py
+++ b/python/paddle/nn/layer/transformer.py
@@ -1396,5 +1396,7 @@ class Transformer(Layer):
                 # [  0.   0.   0.   0.   0.]]
 
         """
-        return paddle.tensor.triu((paddle.ones(
-            (length, length), dtype=paddle.get_default_dtype()) * -np.inf), 1)
+        return paddle.tensor.triu(
+            paddle.full(shape=[length, length],
+                        fill_value=-np.inf,
+                        dtype=paddle.get_default_dtype()), 1)
-- 
GitLab