未验证 提交 e9882514 编写于 作者: A Asthestarsfalll 提交者: GitHub

docs: Fix code example vsplit in docstring (#53558)

上级 10f9249b
...@@ -2098,7 +2098,7 @@ def vsplit(x, num_or_sections, name=None): ...@@ -2098,7 +2098,7 @@ def vsplit(x, num_or_sections, name=None):
# x is a Tensor of shape [8, 6, 7] # x is a Tensor of shape [8, 6, 7]
x = paddle.rand([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(out0.shape) # [4, 6, 7]
print(out1.shape) # [4, 6, 7] print(out1.shape) # [4, 6, 7]
out0, out1, out2 = paddle.vsplit(x, num_or_sections=[1, 3, 4]) out0, out1, out2 = paddle.vsplit(x, num_or_sections=[1, 3, 4])
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册