tf.lookup.KeyValueTensorInitializer

View source on GitHub

Table initializers given keys and values tensors.

View aliases

Compat aliases for migration

See Migration guide for more details.

tf.compat.v1.lookup.KeyValueTensorInitializer

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

Used in the notebooks

Used in the guide	Used in the tutorials
BERT Preprocessing with TF Text Subword tokenizers	Load text Client-efficient large-model federated learning via `federated_select` and sparse aggregation Federated Learning for Text Generation Apache ORC Reader Feature Engineering using TFX Pipeline and TensorFlow Transform

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)

Args
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).

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

Methods

initialize

View source

initialize(
    table
)

Initializes the given table with keys and values tensors.

Args
table	The table to initialize.

Returns
The operation that initializes the table.

Raises
TypeError	when the keys and values data types do not match the table key and value data types.