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]