flatten

    flatten op将输入的多维Tensor展平成2-D Tensor矩阵

    例如：

    参数：

    • x (Variable) - 一个维度数>=axis 的多维Tensor, 数据类型可以为float32，float64，int8，int32或int64。

    • ，则输出张量的Tensor维度为 [1，d0∗d1∗…dn][1，d0∗d1∗…dn] ，默认值为1。

    返回: 一个 2-D Tensor，它包含输入Tensor的数据，但维度发生变化。输入的[0, axis)维将沿axis展平到输出Tensor的0维度，剩余的输入维数展平到输出的1维度。数据类型与输入x相同。

    抛出异常：

    • ValueError: 如果 x 不是一个Variable

    • ValueError: 如果axis的范围不在 [0, rank(x)] 范围内

    代码示例

    2. # x shape is [4, 4, 3]
    3. # out shape is [16, 3]