拆分

Split - 18

版本

  • 名称: Split (GitHub)

  • 域名: main

  • since_version: 18

  • 函数: False

  • support_level: SupportType.COMMON

  • 形状推断: True

此版本的运算符自版本18起可用。

摘要

将一个张量沿着指定的‘axis’分割成一系列张量。 应该指定输入‘split’或属性‘num_outputs’,但不能同时指定两者。 如果指定了属性‘num_outputs’,则张量被分割成大小相等的部分。 如果张量不能被均匀分割成num_outputs,最后一部分将会更小。 如果指定了输入‘split’,它表示分割中每个输出的大小。

属性

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

    在哪个轴上分割。负值表示从后向前计算维度。可接受的范围是[-rank, rank-1],其中r = rank(input)。

  • num_outputs - INT :

    将张量的部分分割成的输出数量。如果张量不能被均匀分割,最后一个块将会更小。

输入

在1和2个输入之间。

  • 输入 (异构) - T:

    要分割的张量

  • split (可选, 异构) - tensor(int64):

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

输出

输出在1到2147483647之间。

  • 输出 (可变参数,异构) - 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) ):

    将输入和输出类型限制为所有张量类型。

拆分 - 13

版本

  • 名称: Split (GitHub)

  • 域名: main

  • since_version: 13

  • 函数: False

  • support_level: SupportType.COMMON

  • 形状推断: True

此版本的运算符自版本13起可用。

摘要

将一个张量沿着指定的‘axis’分割成一个张量列表。可以使用输入‘split’来指定各部分的大小。否则,张量将被分割成大小相等的部分。

属性

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

    在哪个轴上分割。负值表示从后向前计算维度。可接受的范围是[-rank, rank-1],其中r = rank(input)。

输入

在1和2个输入之间。

  • 输入 (异构) - T:

    要分割的张量

  • split (可选, 异构) - tensor(int64):

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

输出

输出在1到2147483647之间。

  • 输出 (可变参数,异构) - 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) ):

    将输入和输出类型限制为所有张量类型。

拆分 - 11

版本

  • 名称: Split (GitHub)

  • 域名: main

  • since_version: 11

  • 函数: False

  • support_level: SupportType.COMMON

  • 形状推断: True

此版本的运算符自版本11起可用。

总结

将一个张量沿着指定的‘axis’分割成一系列张量。可以使用参数‘split’来指定各部分的大小。否则,张量将被分割成大小相等的部分。

属性

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

    在哪个轴上分割。负值表示从后向前计算维度。可接受的范围是[-rank, rank-1],其中r = rank(input)。

  • split - INTS :

    每个输出的长度。值应大于等于0。

输入

  • 输入 (异构) - T:

    要分割的张量

输出

输出在1到2147483647之间。

  • 输出 (可变参数,异构) - 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) ):

    将输入和输出类型限制为所有张量类型。

拆分 - 2

版本

  • 名称: Split (GitHub)

  • 域名: main

  • since_version: 2

  • 函数: False

  • support_level: SupportType.COMMON

  • 形状推断: True

此版本的运算符自版本2起可用。

摘要

将一个张量沿着指定的‘axis’分割成一个张量列表。可以使用参数‘split’来指定各部分的大小。否则,张量将被分割成大小相等的部分。

属性

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

    在哪个轴上分割。

  • split - INTS :

    每个输出的长度

输入

  • 输入 (异构) - T:

    要分割的张量

输出

输出在1到2147483647之间。

  • 输出 (可变参数,异构) - 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) ):

    将输入和输出类型限制为所有张量类型。

分割 - 1

版本

  • 名称: Split (GitHub)

  • 域名: main

  • since_version: 1

  • 函数: False

  • support_level: SupportType.COMMON

  • 形状推断: False

此版本的运算符自版本1起可用。

摘要

将一个张量沿着指定的‘axis’分割成一个张量列表。可以使用参数‘axis’或操作符的可选第二个输入blob来指定分割的长度。否则,张量将被分割成大小相等的部分。

属性

  • axis - INT :

    在哪个轴上分割

  • split - INTS :

    每个输出的长度

输入

在1和2个输入之间。

  • 输入 (异构) - T:

    要分割的张量

  • split (可选, 异构) - T:

    可选的输出长度列表(另请参见参数 'split')

输出

输出在1到2147483647之间。

  • 输出… (可变参数,异构) - T:

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

类型约束

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

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