Python tf.lookup.KeyValueTensorInitializer用法及代码示例

给定 keys 和 values 张量的表初始值设定项。

用法

tf.lookup.KeyValueTensorInitializer(
    keys, values, key_dtype=None, value_dtype=None, name=None
)

参数

keys 键的张量。
values 值的张量。
key_dtype keys 数据类型。当 keys 是 python 数组时使用。
value_dtype values 数据类型。当 values 是 python 数组时使用。
name 操作的名称(可选)。

属性

key_dtype 预期的表键 dtype。
value_dtype 预期的表值 dtype。

keys_tensor = tf.constant(['a', 'b', 'c'])
vals_tensor = tf.constant([7, 8, 9])
input_tensor = tf.constant(['a', 'f'])
init = tf.lookup.KeyValueTensorInitializer(keys_tensor, vals_tensor)
table = tf.lookup.StaticHashTable(
    init,
    default_value=-1)
table.lookup(input_tensor).numpy()
array([ 7, -1], dtype=int32)

相关用法

注：本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.lookup.KeyValueTensorInitializer。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。