tf.lookup.KeyValueTensorInitializer
Инициализаторы таблиц, заданные тензорами keys и values.
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)
| Аргументы |
|---|
keys | Тензор для ключей. |
values | Тензор для значений. |
key_dtype | Тип данных keys. Используется, когда keys — это массив Python. |
value_dtype | Тип данных values. Используется, когда values — это массив Python. |
name | Имя операции (необязательно). |
| Атрибуты |
|---|
key_dtype | Ожидаемый тип данных ключа таблицы. |
value_dtype | Ожидаемый тип данных значения таблицы. |
Методы
initialize
Просмотреть исходный код
initialize(
table
)
Инициализирует данную table тензорами keys и values.
| Аргументы |
|---|
table | Таблица для инициализации. |
| Возвращает |
|---|
| Операция инициализации таблицы. |
| Исключения |
|---|
TypeError | при несовпадении типов данных ключей и значений с типами данных ключа и значения таблицы. |
