SplitToSequence¶

SplitToSequence - 11¶

版本¶

名称: SplitToSequence (GitHub)
域: main
自版本: 11
函数: False
支持级别: SupportType.COMMON
形状推断: True

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

摘要¶

沿着指定的“轴”将张量拆分为一系列张量。可以使用可选参数“split”指定各部分的长度。如果未指定参数split' ，则使用默认标量值 1 作为 split' 的值。 'split' 必须仅包含正数。 'split' 可以是标量（空形状的张量）或一维张量。如果 'split' 是标量，则如果可能， 'input' 将被拆分为大小均为 'split' 的块。如果给定轴 'axis' 上 'input' 的大小不能被 'split' 整除，则仅最后一个块可能小于 'split' 。如果 'split' 是一维张量，则输入张量将被拆分为 'size(split)' 个块， 'axis' 上各部分的长度在 'split' 中指定。在这种情况下， 'split' 中条目的总和必须等于输入张量在 'axis' 上的维度大小。



属性¶

axis - INT（默认值为'0'）
要拆分的轴。负值表示从后开始计算维度。接受的范围是 [-rank, rank-1]。

keepdims - INT（默认值为'1'）
是否保留拆分维度。默认为 1，表示我们保留拆分维度。如果指定了输入“split”，则忽略此属性。




输入¶
1 到 2 个输入。

input（异构） - T
要拆分的张量

split（可选，异构） - I
每个输出的长度。它可以是标量（空形状的张量）或一维张量。所有值都必须 >= 0。




输出¶

output_sequence（异构） - S
拆分后形成的一系列张量的一个或多个输出




类型约束¶

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)）中
将输入类型约束为所有张量类型。

I 在（tensor(int32)、tensor(int64)）中
将拆分大小约束为整数张量。

S 在（seq(tensor(bool))、seq(tensor(complex128))、seq(tensor(complex64))、seq(tensor(double))、seq(tensor(float))、seq(tensor(float16))、seq(tensor(int16))、seq(tensor(int32))、seq(tensor(int64))、seq(tensor(int8))、seq(tensor(string))、seq(tensor(uint16))、seq(tensor(uint32))、seq(tensor(uint64))、seq(tensor(uint8))）中
将输出类型约束为所有张量类型。