first commit (#45254)

933868cd · limingshu · GitHub · 21a242c2 · 933868cd
显示空白变更内容
内联并排

Showing with 4 addition and 2 deletion

python/paddle/nn/layer/transformer.py python/paddle/nn/layer/transformer.py +4 -2

未找到文件。
--- 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(
+        return paddle.tensor.triu(
-            (length, length), dtype=paddle.get_default_dtype()) * -np.inf), 1)
+            paddle.full(shape=[length, length],
+                        fill_value=-np.inf,
+                        dtype=paddle.get_default_dtype()), 1)