Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

tf.lookup.KeyValueTensorInitializer

TensorFlow 1 version View source on GitHub

Table initializers given keys and values tensors.

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

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.