扁平化

Flatten - 23

版本

  • 名称: Flatten (GitHub)

  • 域名: main

  • since_version: 23

  • 函数: False

  • support_level: 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'):

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

输入

  • 输入 (异构) - T:

    一个秩大于等于轴的张量。

输出

  • 输出 (异构) - T:

    一个二维张量，包含输入张量的内容，输入维度直到轴被展平为输出的外部维度，剩余的输入维度被展平为输出的内部维度。

类型约束

  • 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

  • since_version: 21

  • 函数: False

  • support_level: 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'):

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

输入

  • 输入 (异构) - T:

    一个秩大于等于轴的张量。

输出

  • 输出 (异构) - T:

    一个二维张量，包含输入张量的内容，输入维度直到轴被展平为输出的外部维度，剩余的输入维度被展平为输出的内部维度。

类型约束

  • 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

  • since_version: 13

  • 函数: False

  • support_level: 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'):

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

输入

  • 输入 (异构) - T:

    一个秩大于等于轴的张量。

输出

  • 输出 (异构) - T:

    一个二维张量，包含输入张量的内容，输入维度直到轴被展平为输出的外部维度，剩余的输入维度被展平为输出的内部维度。

类型约束

  • 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

  • since_version: 11

  • 函数: False

  • support_level: 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'):

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

输入

  • 输入 (异构) - T:

    一个秩大于等于轴的张量。

输出

  • 输出 (异构) - T:

    一个二维张量，包含输入张量的内容，输入维度直到轴被展平为输出的外部维度，剩余的输入维度被展平为输出的内部维度。

类型约束

  • 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

  • since_version: 9

  • 函数: False

  • support_level: 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'):

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

输入

  • 输入 (异构) - T:

    一个秩大于等于轴的张量。

输出

  • 输出 (异构) - T:

    一个二维张量，包含输入张量的内容，输入维度直到轴被展平为输出的外部维度，剩余的输入维度被展平为输出的内部维度。

类型约束

  • 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

  • since_version: 1

  • 函数: False

  • support_level: 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'):

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

输入

  • 输入 (异构) - T:

    一个秩大于等于轴的张量。

输出

  • 输出 (异构) - T:

    一个二维张量，包含输入张量的内容，输入维度直到轴被展平为输出的外部维度，剩余的输入维度被展平为输出的内部维度。

类型约束

  • T 在 ( tensor(double), tensor(float), tensor(float16) ) 中:

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