提交 5d795306 编写于 作者: M mapingshuo

update example, test=develop

上级 0dee1ba2
...@@ -41,16 +41,18 @@ reshape ...@@ -41,16 +41,18 @@ reshape
import numpy as np import numpy as np
import paddle import paddle
data = np.random.random([2, 4, 6]).astype("float32") x = paddle.rand([2, 4, 6], dtype="float32")
x = paddle.to_tensor(data) positive_four = paddle.full([1], 4, "int32")
positive_four = paddle.fill_constant([1], "int32", 4) out = paddle.reshape(x, [-1, 0, 3, 2])
out_1 = paddle.reshape(x, [-1, 0, 3, 2]) print(out)
# the shape of out_1 is [2,4,3,2]. # the shape is [2,4,3,2].
out_2 = paddle.reshape(x, shape=[positive_four, 12]) out = paddle.reshape(x, shape=[positive_four, 12])
print(out)
# the shape of out_2 is [4, 12]. # the shape of out_2 is [4, 12].
shape_tensor = paddle.to_tensor(np.array([8, 6]).astype("int32")) shape_tensor = paddle.to_tensor(np.array([8, 6]).astype("int32"))
out_3 = paddle.reshape(x, shape=shape_tensor) out = paddle.reshape(x, shape=shape_tensor)
# the shape of out_2 is [8, 6]. print(out)
# the shape is [8, 6].
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册