Flatten¶

paddle.nn. Flatten ( start_axis=1, stop_axis=- 1 ) [源代码] ¶

构造一个 Flatten 类的可调用对象。更多信息请参见代码示例。它实现将一个连续维度的 Tensor 展平成一维 Tensor。

import paddle

inp = paddle.ones([5, 2, 3, 4]).astype('float32')
flatten = paddle.nn.Flatten(start_axis=1, stop_axis=2)
y = flatten(inp)
# y.shape = [5, 6, 4]