paddle 基础函数 flatten

函数文档：https://www.paddlepaddle.org.cn/documentation/docs/zh/api_cn/layers_cn/flatten_cn.html#flatten

 

示例如下：

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 。

 

结果：