From c271f26b18d1e895d2f492345beb3e36f20433a8 Mon Sep 17 00:00:00 2001 From: Sarvesh Shashikumar Date: Tue, 8 Aug 2023 16:27:05 +0530 Subject: [PATCH] Fix error in docstring for flatten function (#56073) Flatten function docstring case 1 displayed incorrect output --- 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 13687977fa2..8053c86cba9 100644 --- a/python/paddle/tensor/manipulation.py +++ b/python/paddle/tensor/manipulation.py @@ -1499,7 +1499,7 @@ def flatten(x, start_axis=0, stop_axis=-1, name=None): end_axis = 2 We get: - Out.shape = (3, 1000 * 100, 2) + Out.shape = (3, 100 * 100, 4) Case 2: -- GitLab