展平

展平 - 23

版本

  • 名称: Flatten (GitHub)

  • 域: main

  • 自版本: 23

  • 函数: False

  • 支持级别: SupportType.COMMON

  • 形状推断: True

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

摘要

将输入张量展平为一个二维矩阵。如果输入张量的形状为 (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

    一个二维张量，包含输入张量的内容，将输入维度中直到 axis 的维度展平到输出的外维度，并将剩余的输入维度展平到输出的内维度。

类型约束

  • T in ( 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 之前的所有张量类型。

展平 - 21

版本

  • 名称: Flatten (GitHub)

  • 域: main

  • 自版本: 21

  • 函数: False

  • 支持级别: SupportType.COMMON

  • 形状推断: True

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

摘要

将输入张量展平为一个二维矩阵。如果输入张量的形状为 (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

    一个二维张量，包含输入张量的内容，将输入维度中直到 axis 的维度展平到输出的外维度，并将剩余的输入维度展平到输出的内维度。

类型约束

  • T in ( 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 之前的所有张量类型。

展平 - 13

版本

  • 名称: Flatten (GitHub)

  • 域: main

  • 自版本: 13

  • 函数: False

  • 支持级别: SupportType.COMMON

  • 形状推断: True

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

摘要

将输入张量展平为一个二维矩阵。如果输入张量的形状为 (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

    一个二维张量，包含输入张量的内容，将输入维度中直到 axis 的维度展平到输出的外维度，并将剩余的输入维度展平到输出的内维度。

类型约束

  • T in ( 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) )

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

展平 - 11

版本

  • 名称: Flatten (GitHub)

  • 域: main

  • since_version: 11

  • 函数: False

  • 支持级别: SupportType.COMMON

  • 形状推断: True

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

摘要

将输入张量展平为一个二维矩阵。如果输入张量的形状为 (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

    一个二维张量，包含输入张量的内容，将输入维度中直到 axis 的维度展平到输出的外维度，并将剩余的输入维度展平到输出的内维度。

类型约束

  • T in ( 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

  • since_version: 9

  • 函数: False

  • 支持级别: SupportType.COMMON

  • 形状推断: True

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

摘要

将输入张量展平为一个二维矩阵。如果输入张量的形状为 (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

    一个二维张量，包含输入张量的内容，将输入维度中直到 axis 的维度展平到输出的外维度，并将剩余的输入维度展平到输出的内维度。

类型约束

  • T in ( 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

  • since_version: 1

  • 函数: False

  • 支持级别: SupportType.COMMON

  • 形状推断: True

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

摘要

将输入张量展平为一个二维矩阵。如果输入张量的形状为 (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

    一个二维张量，包含输入张量的内容，将输入维度中直到 axis 的维度展平到输出的外维度，并将剩余的输入维度展平到输出的内维度。

类型约束

  • T in ( tensor(double), tensor(float), tensor(float16) )

    将输入和输出类型限制为浮点张量。