diff --git a/python/paddle/tensor/manipulation.py b/python/paddle/tensor/manipulation.py index 598ab9d74e14f8c6fe4e2aba407b327a523fd730..4c659f95e0aeb7c0dbd743cf5ef1efc10666b31d 100644 --- a/python/paddle/tensor/manipulation.py +++ b/python/paddle/tensor/manipulation.py @@ -1303,20 +1303,18 @@ def reshape(x, shape, name=None): import numpy as np import paddle - data = np.random.random([2, 4, 6]).astype("float32") - x = paddle.to_tensor(data) - - positive_four = paddle.fill_constant([1], "int32", 4) - - out_1 = paddle.reshape(x, [-1, 0, 3, 2]) - # the shape of out_1 is [2,4,3,2]. - - out_2 = paddle.reshape(x, shape=[positive_four, 12]) + x = paddle.rand([2, 4, 6], dtype="float32") + positive_four = paddle.full([1], 4, "int32") + out = paddle.reshape(x, [-1, 0, 3, 2]) + print(out) + # the shape is [2,4,3,2]. + out = paddle.reshape(x, shape=[positive_four, 12]) + print(out) # the shape of out_2 is [4, 12]. - shape_tensor = paddle.to_tensor(np.array([8, 6]).astype("int32")) - out_3 = paddle.reshape(x, shape=shape_tensor) - # the shape of out_2 is [8, 6]. + out = paddle.reshape(x, shape=shape_tensor) + print(out) + # the shape is [8, 6]. """ return paddle.fluid.layers.reshape(x=x, shape=shape, name=name)