tf.lookup.KeyValueTensorInitializer
Table initializers given keys and values tensors.
tf.lookup.KeyValueTensorInitializer(
keys, values, key_dtype=None, value_dtype=None, name=None
)
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. |