CumSum

CumSum - 14

版本

  • 名稱: CumSum (GitHub)

  • 網域: main

  • since_version: 14

  • function: False

  • 支援層級: SupportType.COMMON

  • 形狀推斷: True

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

摘要

沿著給定軸執行輸入元素的累積總和。依預設,它會執行包含性總和,表示第一個元素會照原樣複製。透過 exclusive 屬性,此行為可以變更為排除第一個元素。它也可以執行軸反方向的總和。為此,將 reverse 屬性設定為 1。

範例

input_x = [1, 2, 3]
axis=0
output = [1, 3, 6]
exclusive=1
output = [0, 1, 3]
exclusive=0
reverse=1
output = [6, 5, 3]
exclusive=1
reverse=1
output = [5, 3, 0]

屬性

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

    如果設定為 1,則會傳回排除性總和,其中不包含最上層元素。換句話說,如果設定為 1,則第 j 個輸出元素會是前 (j-1) 個元素的總和。否則,它會是前 j 個元素的總和。

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

    如果設定為 1,則會以反方向執行總和。

輸入

  • x (異質) - T

    要處理的輸入張量。

  • axis (異質) - T2

    0 維張量。必須在 [-rank(x), rank(x)-1] 的範圍內。負值表示從後面計算維度。

輸出

  • y (異質) - T

    與「x」類型相同的輸出張量,其中包含 x 元素的累積總和

類型約束

  • T in ( tensor(bfloat16), tensor(double), tensor(float), tensor(float16), tensor(int32), tensor(int64), tensor(uint32), tensor(uint64) )

    將輸入和輸出類型限制為高精確度數值張量。

  • T2 in ( tensor(int32), tensor(int64) )

    axis 張量只能是 int32 或 int64

CumSum - 11

版本

  • 名稱: CumSum (GitHub)

  • 網域: main

  • since_version: 11

  • function: False

  • 支援層級: SupportType.COMMON

  • 形狀推斷: True

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

摘要

沿著給定軸執行輸入元素的累積總和。依預設,它會執行包含性總和,表示第一個元素會照原樣複製。透過 exclusive 屬性,此行為可以變更為排除第一個元素。它也可以執行軸反方向的總和。為此,將 reverse 屬性設定為 1。

範例

input_x = [1, 2, 3]
axis=0
output = [1, 3, 6]
exclusive=1
output = [0, 1, 3]
exclusive=0
reverse=1
output = [6, 5, 3]
exclusive=1
reverse=1
output = [5, 3, 0]

屬性

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

    如果設定為 1,則會傳回排除性總和,其中不包含最上層元素。換句話說,如果設定為 1,則第 j 個輸出元素會是前 (j-1) 個元素的總和。否則,它會是前 j 個元素的總和。

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

    如果設定為 1,則會以反方向執行總和。

輸入

  • x (異質) - T

    要處理的輸入張量。

  • axis (異質) - T2

    0 維張量。必須在 [-rank(x), rank(x)-1] 的範圍內。負值表示從後面計算維度。

輸出

  • y (異質) - T

    與「x」類型相同的輸出張量,其中包含 x 元素的累積總和

類型約束

  • T in ( tensor(double), tensor(float), tensor(int32), tensor(int64), tensor(uint32), tensor(uint64) )

    輸入可以是任何張量類型。

  • T2 in ( tensor(int32), tensor(int64) )

    axis 張量只能是 int32 或 int64