分割

分割 - 18

版本

  • 名稱Split (GitHub)

  • 網域main

  • since_version18

  • functionFalse

  • 支援等級SupportType.COMMON

  • 形狀推斷True

此版本的運算子自版本 18 起可用。

摘要

沿著指定的「軸」將張量分割成張量清單。應指定輸入「split」或屬性「num_outputs」,但不能同時指定兩者。如果指定了屬性「num_outputs」,則將張量分割成大小相等的部分。如果張量無法均勻分割成 num_outputs,則最後一塊會比較小。如果指定了輸入「split」,則表示分割中每個輸出的尺寸。

屬性

  • 軸 - INT (預設為 '0')

    要分割的軸。負值表示從後面計算維度。可接受的範圍為 [-rank, rank-1],其中 r = rank(輸入)。

  • num_outputs - INT :

    將張量分割成的輸出部分數量。如果張量無法均勻分割,則最後一塊會比較小。

輸入

介於 1 到 2 個輸入之間。

  • 輸入 (異質) - T

    要分割的張量

  • 分割 (選用,異質) - tensor(int64)

    每個輸出的選用長度。值應 >= 0。這些值的總和必須等於指定「軸」的維度值。

輸出

介於 1 到 2147483647 個輸出之間。

  • 輸出 (可變參數,異質) - T

    分割後形成張量清單的一個或多個輸出

類型約束

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

    將輸入和輸出類型限制為所有張量類型。

分割 - 13

版本

  • 名稱Split (GitHub)

  • 網域main

  • since_version13

  • functionFalse

  • 支援等級SupportType.COMMON

  • 形狀推斷True

此版本的運算子自版本 13 起可用。

摘要

沿著指定的「軸」將張量分割成張量清單。可以使用輸入「split」指定各部分的長度。否則,張量將分割成大小相等的部分。

屬性

  • 軸 - INT (預設為 '0')

    要分割的軸。負值表示從後面計算維度。可接受的範圍為 [-rank, rank-1],其中 r = rank(輸入)。

輸入

介於 1 到 2 個輸入之間。

  • 輸入 (異質) - T

    要分割的張量

  • 分割 (選用,異質) - tensor(int64)

    每個輸出的選用長度。值應 >= 0。這些值的總和必須等於指定「軸」的維度值。

輸出

介於 1 到 2147483647 個輸出之間。

  • 輸出 (可變參數,異質) - T

    分割後形成張量清單的一個或多個輸出

類型約束

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

    將輸入和輸出類型限制為所有張量類型。

分割 - 11

版本

  • 名稱Split (GitHub)

  • 網域main

  • since_version11

  • functionFalse

  • 支援等級SupportType.COMMON

  • 形狀推斷True

此版本的運算子自版本 11 起可用。

摘要

沿著指定的「軸」將張量分割成張量清單。可以使用參數「split」指定各部分的長度。否則,張量將分割成大小相等的部分。

屬性

  • 軸 - INT (預設為 '0')

    要分割的軸。負值表示從後面計算維度。可接受的範圍為 [-rank, rank-1],其中 r = rank(輸入)。

  • split - INTS :

    每個輸出的長度。值應 >= 0。

輸入

  • 輸入 (異質) - T

    要分割的張量

輸出

介於 1 到 2147483647 個輸出之間。

  • 輸出 (可變參數,異質) - T

    分割後形成張量清單的一個或多個輸出

類型約束

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

    將輸入和輸出類型限制為所有張量類型。

分割 - 2

版本

  • 名稱Split (GitHub)

  • 網域main

  • since_version2

  • functionFalse

  • 支援等級SupportType.COMMON

  • 形狀推斷True

此版本的運算子自版本 2 起可用。

摘要

沿著指定的「軸」將張量分割成張量清單。可以使用參數「split」指定各部分的長度。否則,張量將分割成大小相等的部分。

屬性

  • 軸 - INT (預設為 '0')

    要分割的軸向。

  • split - INTS :

    每個輸出的長度

輸入

  • 輸入 (異質) - T

    要分割的張量

輸出

介於 1 到 2147483647 個輸出之間。

  • 輸出 (可變參數,異質) - T

    分割後形成張量清單的一個或多個輸出

類型約束

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

    將輸入和輸出類型限制為所有張量類型。

分割 - 1

版本

  • 名稱Split (GitHub)

  • 網域main

  • since_version: 1

  • functionFalse

  • 支援等級SupportType.COMMON

  • 形狀推斷: False

此版本的運算子自版本 1 起可用。

摘要

沿指定的「軸」將張量分割成張量列表。分割的長度可以使用參數「軸」或運算子的可選第二個輸入 blob 來指定。否則,張量將分割成大小相等的部分。

屬性

  • axis - INT :

    要分割的軸向

  • split - INTS :

    每個輸出的長度

輸入

介於 1 到 2 個輸入之間。

  • 輸入 (異質) - T

    要分割的張量

  • split(可選,異質) - T

    可選的輸出長度列表(另請參閱參數「split」)

輸出

介於 1 到 2147483647 個輸出之間。

  • outputs…(可變參數,異質) - T

    分割後形成張量清單的一個或多個輸出

類型約束

  • T in ( tensor(double), tensor(float), tensor(float16) )

    將輸入類型限制為浮點數張量。