From 1b9aee4f70bce5cdfc4f7c2e90697f452ff88091 Mon Sep 17 00:00:00 2001
From: Liyulingyue <83450930+Liyulingyue@users.noreply.github.com>
Date: Mon, 21 Mar 2022 11:52:48 +0800
Subject: [PATCH] Update manipulation.py (#40722)

---
 python/paddle/tensor/manipulation.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/python/paddle/tensor/manipulation.py b/python/paddle/tensor/manipulation.py
index 9437040eb77..9a9383cee42 100755
--- a/python/paddle/tensor/manipulation.py
+++ b/python/paddle/tensor/manipulation.py
@@ -889,12 +889,20 @@ def stack(x, axis=0, name=None):
             x1 = paddle.to_tensor([[1.0, 2.0]])
             x2 = paddle.to_tensor([[3.0, 4.0]])
             x3 = paddle.to_tensor([[5.0, 6.0]])
+	    
             out = paddle.stack([x1, x2, x3], axis=0)
             print(out.shape)  # [3, 1, 2]
             print(out)
             # [[[1., 2.]],
             #  [[3., 4.]],
             #  [[5., 6.]]]
+	    
+	    out = paddle.stack([x1, x2, x3], axis=-2)
+	    print(out.shape)  # [1, 3, 2]
+	    print(out)
+	    # [[[1., 2.],
+	    #   [3., 4.],
+	    #   [5., 6.]]]
     """
     return layers.stack(x, axis, name)
 
-- 
GitLab