阅读(7.5k) 书签 (0)

计算张量沿轴的累积积

2018-09-25 14:00 更新

tf.cumprod

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

定义在:tensorflow/python/ops/math_ops.py.

参见指南:数学函数>张量扫描

计算张量 x 沿轴的累积积.

默认情况下,此 op 执行包含的 cumprod,这意味着输入的第一个元素与输出的第一个元素相同:

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

通过将专有的 kwarg 设置为 True,将执行一个独占的 cumprod:

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

通过将反向 kwarg 设置为 True, cumprod 将以相反的方向执行:

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

这比使用单独的 tf.reverse 操作更有效.反向和专有的 kwargs 也可以组合在一起:

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

ARGS:

  • x:张量.必须是下列类型之一:float32,float64,int64,int32,uint8,uint16,int16,int8,complex64,complex128,qint8,quint8,qint32,half.
  • axis:int32 类型的张量(默认值:0).
  • exclusive:如果为 True,则执行独占 cumprod.
  • reverse:布尔值(默认值:False).
  • name:操作的名称(可选).

返回:

返回一个张量,与 x 具有相同的类型.