Spec-Zone.ru › TensorFlow 2.9

tf.lookup.StaticHashTable

Просмотреть исходный код на GitHub

Обобщённая хеш-таблица, которая становится неизменяемой после инициализации.

Унаследовано от: TrackableResource

tf.lookup.StaticHashTable(
    initializer, default_value, name=None, experimental_is_anonymous=False
)

Пример использования:

keys_tensor = tf.constant(['a', 'b', 'c'])
vals_tensor = tf.constant([7, 8, 9])
input_tensor = tf.constant(['a', 'f'])
table = tf.lookup.StaticHashTable(
    tf.lookup.KeyValueTensorInitializer(keys_tensor, vals_tensor),
    default_value=-1)
table.lookup(input_tensor).numpy()
array([ 7, -1], dtype=int32)

Или для более питоновского кода:

table[input_tensor].numpy()
array([ 7, -1], dtype=int32)

Результат операции поиска имеет такую же форму, как и аргумент:

input_tensor = tf.constant([['a', 'b'], ['c', 'd']])
table[input_tensor].numpy()
array([[ 7,  8],
       [ 9, -1]], dtype=int32)
Аргументы
initializer Инициализатор таблицы. См. HashTable для поддерживаемых типов ключей и значений.
default_value Значение, используемое, если ключ отсутствует в таблице.
name Имя операции (необязательно).
experimental_is_anonymous Использовать анонимный режим для таблицы (по умолчанию False). В анонимном режиме к ресурсу таблицы можно получить доступ только через дескриптор ресурса. Он не может быть найден по имени. Когда все дескрипторы ресурсов, указывающие на этот ресурс, исчезнут, ресурс будет автоматически удалён.
Атрибуты
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 если ключи разреженные, RaggedTensor если ключи фрагментированы, иначе плотный Tensor.
Исключения
TypeError при keys или default_value не соответствует типам данных таблицы.

size

Просмотреть исходный код

size(
    name=None
)

Вычисляет количество элементов в этой таблице.

Аргументы
name Имя операции (необязательно).
Возвращает
Скалярный тензор, содержащий количество элементов в этой таблице.

__getitem__

Просмотреть исходный код

__getitem__(
    keys
)

Ищет keys в таблице, возвращает соответствующие значения.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/lookup/StaticHashTable

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API