Python tf.math.cumsum用法及代碼示例

計算張量 x 沿 axis 的累積和。

用法

tf.math.cumsum(
    x, axis=0, exclusive=False, reverse=False, name=None
)

默認情況下，此操作執行包含 cumsum，這意味著輸入的第一個元素與輸出的第一個元素相同：例如：

# tf.cumsum([a, b, c])   # [a, a + b, a + b + c]
x = tf.constant([2, 4, 6, 8])
tf.cumsum(x)
<tf.Tensor:shape=(4,), dtype=int32,
numpy=array([ 2,  6, 12, 20], dtype=int32)>

# using varying `axis` values
y = tf.constant([[2, 4, 6, 8], [1,3,5,7]])
tf.cumsum(y, axis=0)
<tf.Tensor:shape=(2, 4), dtype=int32, numpy=
array([[ 2,  4,  6,  8],
       [ 3,  7, 11, 15]], dtype=int32)>
tf.cumsum(y, axis=1)
<tf.Tensor:shape=(2, 4), dtype=int32, numpy=
array([[ 2,  6, 12, 20],
       [ 1,  4,  9, 16]], dtype=int32)>

通過將 exclusive kwarg 設置為 True ，將執行獨占 cumsum：

# tf.cumsum([a, b, c], exclusive=True)  => [0, a, a + b]
x = tf.constant([2, 4, 6, 8])
tf.cumsum(x, exclusive=True)
<tf.Tensor:shape=(4,), dtype=int32,
numpy=array([ 0,  2,  6, 12], dtype=int32)>

通過將 reverse kwarg 設置為 True ，cumsum 以相反的方向執行：

# tf.cumsum([a, b, c], reverse=True)  # [a + b + c, b + c, c]
x = tf.constant([2, 4, 6, 8])
tf.cumsum(x, reverse=True)
<tf.Tensor:shape=(4,), dtype=int32,
numpy=array([20, 18, 14,  8], dtype=int32)>

這比使用單獨的 tf.reverse 操作更有效。 reverse 和 exclusive kwargs 也可以組合：

# tf.cumsum([a, b, c], exclusive=True, reverse=True)  # [b + c, c, 0]
x = tf.constant([2, 4, 6, 8])
tf.cumsum(x, exclusive=True, reverse=True)
<tf.Tensor:shape=(4,), dtype=int32,
numpy=array([18, 14,  8,  0], dtype=int32)>