计算张量沿轴的累积积
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 具有相同的类型.