contrib.lookup.InitializableLookupTableBase

tf.contrib.lookup.InitializableLookupTableBase

class tf.contrib.lookup.InitializableLookupTableBase

Defined in tensorflow/contrib/lookup/lookup_ops.py.

Initializable lookup table interface.

An initializable lookup tables persist across different steps.

Properties

default_value

The default value of the table.

init

The table initialization op.

key_dtype

The table key dtype.

name

The name of the table.

table_ref

Get the underlying table reference.

value_dtype

The table value dtype.

Methods

__init__

__init__(
    table_ref,
    default_value,
    initializer
)

Construct a table object from a table reference.

If requires a table initializer object (subclass of TableInitializerBase). It provides the table key and value types, as well as the op to initialize the table. The caller is responsible to execute the initialization op.

Args:

• table_ref: The table reference, i.e. the output of the lookup table ops.
• default_value: The value to use if a key is missing in the table.
• initializer: The table initializer to use.

check_table_dtypes

check_table_dtypes(
    key_dtype,
    value_dtype
)

Check that the given key_dtype and value_dtype matches the table dtypes.

Args:

• key_dtype: The key data type to check.
• value_dtype: The value data type to check.

Raises:

• TypeError: when 'key_dtype' or 'value_dtype' doesn't match the table data types.

lookup

lookup(
    keys,
    name=None
)

Looks up keys in a table, outputs the corresponding values.

The default_value is used for keys not present in the table.

Args:

• keys: Keys to look up. May be either a SparseTensor or dense Tensor.
• name: A name for the operation (optional).

Returns:

A SparseTensor if keys are sparse, otherwise a dense Tensor.

Raises:

• TypeError: when keys or default_value doesn't match the table data types.

size

size(name=None)

Compute the number of elements in this table.

Args:

• name: A name for the operation (optional).

Returns:

A scalar tensor containing the number of elements in this table.