Flatten

class paddle.nn. Flatten ( start_axis=1, stop_axis=- 1 )

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

参数

  • start_axis (int,可选) - 展开的起始维度,默认值为 1。

  • stop_axis (int,可选) - 展开的结束维度,默认值为-1。

返回

无。

代码示例

COPY-FROM: paddle.nn.Flatten