Python tf.data.Dataset.reduce用法及代碼示例

用法

reduce(
    initial_state, reduce_func, name=None
)

initial_state 表示轉換初始狀態的元素。
reduce_func 將 (old_state, input_element) 映射到 new_state 的函數。它必須接受兩個參數並返回一個新元素 new_state 的結構必須與 initial_state 的結構匹配。
name (可選。) tf.data 操作的名稱。

將輸入數據集縮減為單個元素。

轉換在輸入數據集的每個元素上連續調用reduce_func，直到數據集用盡，在其內部狀態中聚合信息。 initial_state 參數用於初始狀態，最終狀態作為結果返回。

tf.data.Dataset.range(5).reduce(np.int64(0), lambda x, _:x + 1).numpy()
5
tf.data.Dataset.range(5).reduce(np.int64(0), lambda x, y:x + y).numpy()
10

相關用法

注：本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.data.Dataset.reduce。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。