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

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

用法

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

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

tf.math.cumprod([a, b, c])  # [a, a * b, a * b * c]

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

tf.math.cumprod([a, b, c], exclusive=True)  # [1, a, a * b]

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

tf.math.cumprod([a, b, c], reverse=True)  # [a * b * c, b * c, c]

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

tf.math.cumprod([a, b, c], exclusive=True, reverse=True)  # [b * c, c, 1]