diff --git a/python/paddle/tensor/manipulation.py b/python/paddle/tensor/manipulation.py index 13687977fa23cbda3f209b368aa32e6597728474..8053c86cba9de98f66c148564d7edd157b774c37 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: