From e988251404a96a8d36d628b49d6002d28efc795b Mon Sep 17 00:00:00 2001 From: Asthestarsfalll <72954905+Asthestarsfalll@users.noreply.github.com> Date: Mon, 8 May 2023 14:53:06 +0800 Subject: [PATCH] docs: Fix code example vsplit in docstring (#53558) --- python/paddle/tensor/manipulation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/paddle/tensor/manipulation.py b/python/paddle/tensor/manipulation.py index 5c4a029711b..d0427886be2 100644 --- a/python/paddle/tensor/manipulation.py +++ b/python/paddle/tensor/manipulation.py @@ -2098,7 +2098,7 @@ def vsplit(x, num_or_sections, name=None): # x is a Tensor of shape [8, 6, 7] x = paddle.rand([8, 6, 7]) - out0, out1, out2 = paddle.vsplit(x, num_or_sections=2) + out0, out1 = paddle.vsplit(x, num_or_sections=2) print(out0.shape) # [4, 6, 7] print(out1.shape) # [4, 6, 7] out0, out1, out2 = paddle.vsplit(x, num_or_sections=[1, 3, 4]) -- GitLab