当前位置: 首页>>代码示例 >>用法及示例精选 >>正文

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。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。