Spec-Zone.ru › TensorFlow

tf.lookup.StaticHashTable

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

Наследуется от: TrackableResource

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

Используется в блокнотах

Используется в учебниках
  • Федеральное обучение для генерации текста
  • Эффективное для клиентов федеральное обучение с большими моделями с помощью `federated_select` и разреженной агрегации
  • Инженерия признаков с помощью конвейера TFX и TensorFlow Transform
  • Предварительная обработка данных с помощью TensorFlow Transform

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

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/api_docs/python/tf/lookup/StaticHashTable

Spec-Zone.ru

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