計算張量沿軸的累積積

2018-09-25 14:00 更新

tf.cumprod

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

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

參見指南:數(shù)學函數(shù)>張量掃描

計算張量 x 沿軸的累積積.

默認情況下,此 op 執(zhí)行包含的 cumprod,這意味著輸入的第一個元素與輸出的第一個元素相同:

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

通過將專有的 kwarg 設(shè)置為 True,將執(zhí)行一個獨占的 cumprod:

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

通過將反向 kwarg 設(shè)置為 True, cumprod 將以相反的方向執(zhí)行:

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,則執(zhí)行獨占 cumprod.
  • reverse:布爾值(默認值:False).
  • name:操作的名稱(可選).

返回:

返回一個張量,與 x 具有相同的類型.

以上內(nèi)容是否對您有幫助:
在線筆記
App下載
App下載

掃描二維碼

下載編程獅App

公眾號
微信公眾號

編程獅公眾號