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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。