當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文

Python tf.data.experimental.RandomDataset.scan用法及代碼示例


用法

scan(
    initial_state, scan_func, name=None
)

參數

  • initial_state 張量的嵌套結構，表示累加器的初始狀態。
  • scan_func 將 (old_state, input_element) 映射到 (new_state, output_element) 的函數。它必須接受兩個參數並返回一對張量的嵌套結構。 new_state 必須與 initial_state 的結構匹配。
  • name (可選。) tf.data 操作的名稱。

返回

  • 一個Dataset。

一種跨輸入數據集掃描函數的轉換。

此轉換是 tf.data.Dataset.map 的有狀態相對。除了將scan_func 映射到輸入數據集的元素之外，scan() 還會累積一個或多個狀態張量，其初始值為 initial_state 。

dataset = tf.data.Dataset.range(10)
initial_state = tf.constant(0, dtype=tf.int64)
scan_func = lambda state, i:(state + i, state + i)
dataset = dataset.scan(initial_state=initial_state, scan_func=scan_func)
list(dataset.as_numpy_iterator())
[0, 1, 3, 6, 10, 15, 21, 28, 36, 45]

相關用法


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