SplitToSequence

SplitToSequence - 11

版本

• 名稱: SplitToSequence (GitHub)

• 網域: main

• since_version: 11

• function: False

• 支援層級: SupportType.COMMON

• 形狀推論: True

此版本的運算子自版本 11 起提供。

摘要

沿著指定的 'axis' 將張量分割成張量序列。各部分的長度可以使用可選參數 'split' 來指定。如果未指定引數 split' ，則預設純量值 1 會用作 split' 的值。'split' 必須僅包含正數。'split' 可以是純量 (空形狀的張量) 或 1 維張量。如果 'split' 是純量，則如果可能，'input' 將會被分割成大小為 'split' 的區塊。如果沿著給定軸 'axis' 的 'input' 大小無法被 'split' 整除，則最後一個區塊可能會小於 'split'。如果 'split' 是 1 維張量，則輸入張量會被分割成 'size(split)' 個區塊，且 'axis' 上各部分的長度會在 'split' 中指定。在此情況下，'split' 中條目的總和必須等於 'axis' 上輸入張量的維度大小。

屬性

• axis - INT (預設值為 '0')

  要分割的軸。負值表示從後方開始計算維度。接受範圍為 [-rank, rank-1]。

• keepdims - INT (預設值為 '1')

  是否保留分割維度。預設值為 1，表示我們保留分割維度。如果指定輸入 'split'，則會忽略此屬性。

輸入

介於 1 到 2 個輸入之間。

• input (異質) - T

  要分割的張量

• split (可選，異質) - I

  每個輸出的長度。它可以是純量（空形狀的張量）或 1 維張量。所有值必須 >= 0。

輸出

• output_sequence (異質) - S

  分割後形成張量序列的一個或多個輸出

類型約束

• T in ( 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 in ( tensor(int32), tensor(int64) )

  將分割大小約束為整數張量。

• S in ( 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)) )

  將輸出類型約束為所有張量類型。