ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

tf.lookup.KeyValueTensorInitializer

Table initializers given keys and values tensors.

Used in the notebooks

Used in the guide Used in the tutorials
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)

keys The tensor for the keys.
values The tensor for the values.
key_dtype The keys data type. Used when keys is a python array.
value_dtype The values data type. Used when values is a python array.
name A name for the operation (optional).

key_dtype The expected table key dtype.
value_dtype The expected table value dtype.

Methods

initialize

View source

Initializes the given table with keys and values tensors.

Args
table The table to initialize.

Returns