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

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


從 tf.data.Dataset 創建表初始值設定項。

用法

tf.data.experimental.DatasetInitializer(
    dataset
)

參數

  • dataset 一個生成標量元組的tf.data.Dataset 對象。第一個標量被視為鍵，第二個被視為值。

屬性

  • dataset 一個生成標量元組的tf.data.Dataset 對象。第一個標量被視為鍵，第二個被視為值。
  • key_dtype 預期的表鍵 dtype。
  • value_dtype 預期的表值 dtype。

示例用法：

keys = tf.data.Dataset.range(100)
values = tf.data.Dataset.range(100).map(
    lambda x:tf.strings.as_string(x * 2))
ds = tf.data.Dataset.zip((keys, values))
init = tf.data.experimental.DatasetInitializer(ds)
table = tf.lookup.StaticHashTable(init, "")
table.lookup(tf.constant([0, 1, 2], dtype=tf.int64)).numpy()
array([b'0', b'2', b'4'], dtype=object)

如果dataset 不符合規範，則引發：ValueError。

相關用法


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