import paddle.fluid as fluid
import numpy as np
x = fluid.layers.data(name="x", shape=[3, 2, 3], append_batch_size=False, dtype="float32")
out = fluid.layers.flatten(x=x, axis=1) # out shape is [3, 6]
place = fluid.CPUPlace()
exe = fluid.Executor(place)
exe.run(fluid.default_startup_program())
np_x = np.random.random(size=(3, 2, 3)).astype('float32')
print(np_x)
output = exe.run(feed={"x": np_x}, fetch_list = [out])
print(output)
官方文档 https://www.paddlepaddle.org.cn/documentation/docs/zh/api_cn/layers_cn/data_cn.html#data 中不推荐使用 paddle.fluid.layers.data ,因其在之后的版本中会被删除。请使用 paddle.fluid.data 。