SplitToSequence

SplitToSequence - 11

版本

• 名称: SplitToSequence (GitHub)

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

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