flatten

    flatten op 根据给定的start_axis 和 stop_axis 将连续的维度展平。

    请注意，在动态图模式下，输出Tensor将与输入Tensor共享数据，并且没有Tensor数据拷贝的过程。 如果不希望输入与输出共享数据，请使用 Tensor.clone ，例如 flatten_clone_x = x.flatten().clone() 。

    例如：

    参数：

    • start_axis (int) - flatten展开的起始维度。

    • stop_axis (int) - flatten展开的结束维度。

    • name (str，可选) - 具体用法请参见 Name ，一般无需设置，默认值为None。

    返回: Tensor, 一个 Tensor，它包含输入Tensor的数据，但维度发生变化。输入将按照给定的start_axis 和 stop_axis展开。数据类型与输入x相同。

    抛出异常：

    • ValueError: 如果 x 不是一个Tensor

    代码示例

    1. import paddle
    4. x = paddle.arange(end=image_shape[0] * image_shape[1] * image_shape[2] * image_shape[3])
    5. img = paddle.reshape(x, image_shape) / 100
    7. # out shape is [2, 12, 4]
    9. # 在动态图模式下，输出out与输入img共享数据