Split

Split - 18

版本

  • 名称: Split (GitHub)

  • 域: main

  • 自版本: 18

  • 函数: False

  • 支持级别: SupportType.COMMON

  • 形状推断: True

此版本的操作符自版本 18起可用。

摘要

将张量沿指定的“轴”分割成张量列表。应该指定输入“split”或属性“num_outputs”，但不能同时指定两者。如果指定了属性“num_outputs”，则张量将被分成大小相等的部分。如果张量不能均匀地分成 num_outputs，则最后一个块将更小。如果指定了输入“split”，则它指示 split 中每个输出的大小。

属性

  • axis - INT (默认值为 '0')

    要分割的轴。负值表示从后部计算维度。接受的范围是 [-rank, rank-1]，其中 r = rank(input)。

  • num_outputs - INT :

    将张量分割成的输出数量。如果张量不能均匀地分割，则最后一个块将更小。

输入

1 到 2 个输入之间。

  • input (异构) - T

    要分割的张量

  • split (可选，异构) - tensor(int64)

    每个输出的可选长度。值应 >= 0。值的总和必须等于指定的“轴”处的 dim 值。

输出

1 到 2147483647 个输出之间。

  • outputs (可变，异构) - T

    分割后形成张量列表的一个或多个输出

类型约束

  • 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) )

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

Split - 13

版本

  • 名称: Split (GitHub)

  • 域: main

  • 自版本: 13

  • 函数: False

  • 支持级别: SupportType.COMMON

  • 形状推断: True

此版本的操作符自版本 13起可用。

摘要

将张量沿指定的“轴”分割成张量列表。可以使用输入“split”指定各部分的长度。否则，张量将被分成大小相等的部分。

属性

  • axis - INT (默认值为 '0')

    要分割的轴。负值表示从后部计算维度。接受的范围是 [-rank, rank-1]，其中 r = rank(input)。

输入

1 到 2 个输入之间。

  • input (异构) - T

    要分割的张量

  • split (可选，异构) - tensor(int64)

    每个输出的可选长度。值应 >= 0。值的总和必须等于指定的“轴”处的 dim 值。

输出

1 到 2147483647 个输出之间。

  • outputs (可变，异构) - T

    分割后形成张量列表的一个或多个输出

类型约束

  • 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) )

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

Split - 11

版本

  • 名称: Split (GitHub)

  • 域: main

  • 自版本: 11

  • 函数: False

  • 支持级别: SupportType.COMMON

  • 形状推断: True

此版本的操作符自版本 11起可用。

摘要

将张量沿指定的“轴”分割成张量列表。可以使用参数“split”指定各部分的长度。否则，张量将被分成大小相等的部分。

属性

  • axis - INT (默认值为 '0')

    要分割的轴。负值表示从后部计算维度。接受的范围是 [-rank, rank-1]，其中 r = rank(input)。

  • split - INTS :

    每个输出的长度。值应 >= 0。

输入

  • input (异构) - T

    要分割的张量

输出

1 到 2147483647 个输出之间。

  • outputs (可变，异构) - T

    分割后形成张量列表的一个或多个输出

类型约束

  • 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) )

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

Split - 2

版本

  • 名称: Split (GitHub)

  • 域: main

  • 自版本: 2

  • 函数: False

  • 支持级别: SupportType.COMMON

  • 形状推断: True

此版本的操作符自版本 2起可用。

摘要

将张量沿指定的“轴”分割成张量列表。可以使用参数“split”指定各部分的长度。否则，张量将被分成大小相等的部分。

属性

  • axis - INT (默认值为 '0')

    要分割的轴。

  • split - INTS :

    每个输出的长度

输入

  • input (异构) - T

    要分割的张量

输出

1 到 2147483647 个输出之间。

  • outputs (可变，异构) - T

    分割后形成张量列表的一个或多个输出

类型约束

  • 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) )

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

Split - 1

版本

  • 名称: Split (GitHub)

  • 域: main

  • since_version: 1

  • 函数: False

  • 支持级别: SupportType.COMMON

  • shape inference: False

该版本的操作符从版本1开始可用。

概述

将张量沿指定的“轴”分割成张量列表。分割长度可以通过参数“轴”或可选的第二个输入 blob 指定。否则，张量将被分割成大小相等的多个部分。

属性

  • axis - INT :

    要分割的轴

  • split - INTS :

    每个输出的长度

输入

1 到 2 个输入之间。

  • input (异构) - T

    要分割的张量

  • split (可选，异构) - T

    可选的输出长度列表（参见参数“split”）

输出

1 到 2147483647 个输出之间。

  • outputs… (变长，异构) - T

    分割后形成张量列表的一个或多个输出

类型约束

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

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