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

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