SplitToSequence¶
SplitToSequence - 11¶
版本¶
域:
main
起始版本:
11
函数:
False
支持级别:
SupportType.COMMON
形状推断:
True
此版本运算符自 版本 11 起可用。
概述¶
沿指定的“axis”将张量分割为一系列张量。可以使用可选参数“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))
)将输出类型约束为所有张量类型。