tf.lookup.KeyValueTensorInitializer

TensorFlow 2 version View source on GitHub

Table initializers given keys and values tensors.

Inherits From: TableInitializerBase

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

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

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.