Flatten¶

Flatten - 23¶

版本¶

  • **名称**: Flatten (GitHub)

  • **领域**: main

  • **起始版本**: 23

  • **函数**: False

  • **支持级别**: SupportType.COMMON

  • **形状推断**: True

此算子版本**自版本 23** 起可用。

摘要¶

将输入张量展平为 2D 矩阵。如果输入张量的形状为 (d_0, d_1, … d_n)，则输出的形状将为 (d_0 X d_1 … d_(axis-1), d_axis X d_(axis+1) … X dn)。

属性¶

  • **axis - INT** (默认值为 '1')¶

    指定将哪些输入维度（不包含）展平到输出的外部维度。axis 的值必须在 [-r, r] 范围内，其中 r 是输入张量的秩。负值表示从后向前计数维度。当 axis = 0 时，输出张量的形状为 (1, (d_0 X d_1 … d_n)，其中输入张量的形状为 (d_0, d_1, … d_n)。

输入¶

  • **input** (异构) - **T**¶

    秩 >= axis 的张量。

输出¶

  • **output** (异构) - **T**¶

    一个 2D 张量，包含输入张量的内容，其中 up to axis（不包含）的输入维度展平到输出的外部维度，剩余的输入维度展平到输出的内部维度。

类型约束¶

  • **T** 在 ( tensor(bfloat16), tensor(bool), tensor(complex128), tensor(complex64), tensor(double), tensor(float), tensor(float16), tensor(float4e2m1), tensor(float8e4m3fn), tensor(float8e4m3fnuz), tensor(float8e5m2), tensor(float8e5m2fnuz), tensor(int16), tensor(int32), tensor(int4), tensor(int64), tensor(int8), tensor(string), tensor(uint16), tensor(uint32), tensor(uint4), tensor(uint64), tensor(uint8) ) 中¶

    将输入和输出约束为 IRv10 及之前版本的所有张量类型。

Flatten - 21¶

版本¶

  • **名称**: Flatten (GitHub)

  • **领域**: main

  • **起始版本**: 21

  • **函数**: False

  • **支持级别**: SupportType.COMMON

  • **形状推断**: True

此算子版本**自版本 21** 起可用。

摘要¶

将输入张量展平为 2D 矩阵。如果输入张量的形状为 (d_0, d_1, … d_n)，则输出的形状将为 (d_0 X d_1 … d_(axis-1), d_axis X d_(axis+1) … X dn)。

属性¶

  • **axis - INT** (默认值为 '1')¶

    指定将哪些输入维度（不包含）展平到输出的外部维度。axis 的值必须在 [-r, r] 范围内，其中 r 是输入张量的秩。负值表示从后向前计数维度。当 axis = 0 时，输出张量的形状为 (1, (d_0 X d_1 … d_n)，其中输入张量的形状为 (d_0, d_1, … d_n)。

输入¶

  • **input** (异构) - **T**¶

    秩 >= axis 的张量。

输出¶

  • **output** (异构) - **T**¶

    一个 2D 张量，包含输入张量的内容，其中 up to axis（不包含）的输入维度展平到输出的外部维度，剩余的输入维度展平到输出的内部维度。

类型约束¶

  • **T** 在 ( tensor(bfloat16), tensor(bool), tensor(complex128), tensor(complex64), tensor(double), tensor(float), tensor(float16), tensor(float8e4m3fn), tensor(float8e4m3fnuz), tensor(float8e5m2), tensor(float8e5m2fnuz), tensor(int16), tensor(int32), tensor(int4), tensor(int64), tensor(int8), tensor(string), tensor(uint16), tensor(uint32), tensor(uint4), tensor(uint64), tensor(uint8) ) 中¶

    将输入和输出约束为 IRv10 及之前版本的所有张量类型。

Flatten - 13¶

版本¶

  • **名称**: Flatten (GitHub)

  • **领域**: main

  • **起始版本**: 13

  • **函数**: False

  • **支持级别**: SupportType.COMMON

  • **形状推断**: True

此算子版本**自版本 13** 起可用。

摘要¶

将输入张量展平为 2D 矩阵。如果输入张量的形状为 (d_0, d_1, … d_n)，则输出的形状将为 (d_0 X d_1 … d_(axis-1), d_axis X d_(axis+1) … X dn)。

属性¶

  • **axis - INT** (默认值为 '1')¶

    指定将哪些输入维度（不包含）展平到输出的外部维度。axis 的值必须在 [-r, r] 范围内，其中 r 是输入张量的秩。负值表示从后向前计数维度。当 axis = 0 时，输出张量的形状为 (1, (d_0 X d_1 … d_n)，其中输入张量的形状为 (d_0, d_1, … d_n)。

输入¶

  • **input** (异构) - **T**¶

    秩 >= axis 的张量。

输出¶

  • **output** (异构) - **T**¶

    一个 2D 张量，包含输入张量的内容，其中 up to axis（不包含）的输入维度展平到输出的外部维度，剩余的输入维度展平到输出的内部维度。

类型约束¶

  • **T** 在 ( tensor(bfloat16), tensor(bool), tensor(complex128), tensor(complex64), tensor(double), tensor(float), tensor(float16), tensor(int16), tensor(int32), tensor(int64), tensor(int8), tensor(string), tensor(uint16), tensor(uint32), tensor(uint64), tensor(uint8) ) 中¶

    将输入和输出约束为所有张量类型。

Flatten - 11¶

版本¶

  • **名称**: Flatten (GitHub)

  • **领域**: main

  • **起始版本**: 11

  • **函数**: False

  • **支持级别**: SupportType.COMMON

  • **形状推断**: True

此算子版本**自版本 11** 起可用。

摘要¶

将输入张量展平为 2D 矩阵。如果输入张量的形状为 (d_0, d_1, … d_n)，则输出的形状将为 (d_0 X d_1 … d_(axis-1), d_axis X d_(axis+1) … X dn)。

属性¶

  • **axis - INT** (默认值为 '1')¶

    指定将哪些输入维度（不包含）展平到输出的外部维度。axis 的值必须在 [-r, r] 范围内，其中 r 是输入张量的秩。负值表示从后向前计数维度。当 axis = 0 时，输出张量的形状为 (1, (d_0 X d_1 … d_n)，其中输入张量的形状为 (d_0, d_1, … d_n)。

输入¶

  • **input** (异构) - **T**¶

    秩 >= axis 的张量。

输出¶

  • **output** (异构) - **T**¶

    一个 2D 张量，包含输入张量的内容，其中 up to axis（不包含）的输入维度展平到输出的外部维度，剩余的输入维度展平到输出的内部维度。

类型约束¶

  • **T** 在 ( tensor(bool), tensor(complex128), tensor(complex64), tensor(double), tensor(float), tensor(float16), tensor(int16), tensor(int32), tensor(int64), tensor(int8), tensor(string), tensor(uint16), tensor(uint32), tensor(uint64), tensor(uint8) ) 中¶

    将输入和输出约束为所有张量类型。

Flatten - 9¶

版本¶

  • **名称**: Flatten (GitHub)

  • **领域**: main

  • **起始版本**: 9

  • **函数**: False

  • **支持级别**: SupportType.COMMON

  • **形状推断**: True

此算子版本**自版本 9** 起可用。

摘要¶

将输入张量展平为 2D 矩阵。如果输入张量的形状为 (d_0, d_1, … d_n)，则输出的形状将为 (d_0 X d_1 … d_(axis-1), d_axis X d_(axis+1) … X dn)。

属性¶

  • **axis - INT** (默认值为 '1')¶

    指定将哪些输入维度（不包含）展平到输出的外部维度。axis 的值必须在 [0, R] 范围内，其中 R 是输入张量的秩。当 axis = 0 时，输出张量的形状为 (1, (d_0 X d_1 … d_n)，其中输入张量的形状为 (d_0, d_1, … d_n)。

输入¶

  • **input** (异构) - **T**¶

    秩 >= axis 的张量。

输出¶

  • **output** (异构) - **T**¶

    一个 2D 张量，包含输入张量的内容，其中 up to axis（不包含）的输入维度展平到输出的外部维度，剩余的输入维度展平到输出的内部维度。

类型约束¶

  • **T** 在 ( tensor(bool), tensor(complex128), tensor(complex64), tensor(double), tensor(float), tensor(float16), tensor(int16), tensor(int32), tensor(int64), tensor(int8), tensor(string), tensor(uint16), tensor(uint32), tensor(uint64), tensor(uint8) ) 中¶

    将输入和输出约束为所有张量类型。

Flatten - 1¶

版本¶

  • **名称**: Flatten (GitHub)

  • **领域**: main

  • **起始版本**: 1

  • **函数**: False

  • **支持级别**: SupportType.COMMON

  • **形状推断**: True

此算子版本**自版本 1** 起可用。

摘要¶

将输入张量展平为 2D 矩阵。如果输入张量的形状为 (d_0, d_1, … d_n)，则输出的形状将为 (d_0 X d_1 … d_(axis-1), d_axis X d_(axis+1) … X dn)。

属性¶

  • **axis - INT** (默认值为 '1')¶

    指定将哪些输入维度（不包含）展平到输出的外部维度。axis 的值必须在 [0, R] 范围内，其中 R 是输入张量的秩。当 axis = 0 时，输出张量的形状为 (1, (d_0 X d_1 … d_n)，其中输入张量的形状为 (d_0, d_1, … d_n)。

输入¶

  • **input** (异构) - **T**¶

    秩 >= axis 的张量。

输出¶

  • **output** (异构) - **T**¶

    一个 2D 张量，包含输入张量的内容，其中 up to axis（不包含）的输入维度展平到输出的外部维度，剩余的输入维度展平到输出的内部维度。

类型约束¶

  • **T** 在 ( tensor(double), tensor(float), tensor(float16) ) 中¶

    将输入和输出类型约束为浮点张量。