tf.lookup.StaticHashTable
Обобщённая хеш-таблица, которая становится неизменяемой после инициализации.
tf.lookup.StaticHashTable(
initializer, default_value, name=None
)
Пример использования:
keys_tensor = tf.constant([1, 2])
vals_tensor = tf.constant([3, 4])
input_tensor = tf.constant([1, 5])
table = tf.lookup.StaticHashTable(
tf.lookup.KeyValueTensorInitializer(keys_tensor, vals_tensor), -1)
print(table.lookup(input_tensor))
| Аргументы |
|---|
initializer | Инициализатор таблицы. См. HashTable ядро для поддерживаемых типов ключей и значений. |
default_value | Значение по умолчанию для отсутствующих ключей. |
name | Имя операции (необязательно). |
| Атрибуты |
|---|
default_value | Значение по умолчанию таблицы. |
key_dtype | Тип данных ключей таблицы. |
name | Имя таблицы. |
resource_handle | Возвращает дескриптор ресурса, связанный с этим ресурсом. |
value_dtype | Тип данных значений таблицы. |
Методы
export
Просмотреть исходный код
export(
name=None
)
Возвращает тензоры всех ключей и значений в таблице.
| Аргументы |
|---|
name | Имя операции (необязательно). |
| Возвращаемое значение |
|---|
| Пара тензоров, где первый тензор содержит все ключи, а второй — все значения в таблице. |
lookup
Просмотреть исходный код
lookup(
keys, name=None
)
Ищет keys в таблице, возвращает соответствующие значения.
Значение default_value используется для ключей, отсутствующих в таблице.
| Аргументы |
|---|
keys | Ключи для поиска. Может быть либо SparseTensor , либо плотный Tensor. |
name | Имя операции (необязательно). |
| Возвращаемое значение |
|---|
SparseTensor , если ключи разреженные, в противном случае плотный Tensor. |
| Исключения |
|---|
TypeError | если keys или default_value не соответствуют типам данных таблицы. |
size
Просмотреть исходный код
size(
name=None
)
Вычисляет количество элементов в этой таблице.
| Аргументы |
|---|
name | Имя операции (необязательно). |
| Возвращаемое значение |
|---|
| Скалярный тензор, содержащий количество элементов в этой таблице. |
