Spec-Zone.ru › TensorFlow

tf.lookup.experimental.MutableHashTable

Реализация обобщенной изменяемой хэш-таблицы.

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

Просмотр псевдонимов

Псевдонимы совместимости для миграции

Дополнительные сведения см. в руководстве по миграции.

tf.compat.v1.lookup.experimental.MutableHashTable

tf.lookup.experimental.MutableHashTable(
    key_dtype,
    value_dtype,
    default_value,
    name='MutableHashTable',
    checkpoint=True,
    experimental_is_anonymous=False
)

Данные можно вставить, вызвав метод insert, и удалить, вызвав метод remove. Он не поддерживает инициализацию с помощью метода init.

MutableHashTable требует дополнительной памяти во время операций сохранения и восстановления для создания временных тензоров ключей и значений.

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

table = tf.lookup.experimental.MutableHashTable(key_dtype=tf.string,
                                                value_dtype=tf.int64,
                                                default_value=-1)
keys_tensor = tf.constant(['a', 'b', 'c'])
vals_tensor = tf.constant([7, 8, 9], dtype=tf.int64)
input_tensor = tf.constant(['a', 'f'])
table.insert(keys_tensor, vals_tensor)
table.lookup(input_tensor).numpy()
array([ 7, -1])
table.remove(tf.constant(['c']))
table.lookup(keys_tensor).numpy()
array([ 7, 8, -1])
sorted(table.export()[0].numpy())
[b'a', b'b']
sorted(table.export()[1].numpy())
[7, 8]
Аргументы
key_dtype тип тензоров ключей.
value_dtype тип тензоров значений.
default_value значение по умолчанию для отсутствующих ключей в таблице.
name имя операции (необязательно).
checkpoint если True, содержимое таблицы сохраняется и восстанавливается из контрольных точек. Если shared_name пусто для таблицы контрольной точки, она используется совместно с именем узла таблицы.
experimental_is_anonymous Использовать анонимный режим для таблицы (по умолчанию False). В анонимном режиме доступ к ресурсу таблицы возможен только через дескриптор ресурса. Его нельзя найти по имени. При исчезновении всех дескрипторов ресурсов, указывающих на этот ресурс, ресурс будет автоматически удален.
Исключения
ValueError Если checkpoint=True и имя не указано.
Атрибуты
key_dtype Тип ключа таблицы.
name Имя таблицы.
resource_handle Возвращает дескриптор ресурса, связанный с этим ресурсом.
value_dtype Тип значения таблицы.

Методы

export

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

export(
    name=None
)

Возвращает тензоры всех ключей и значений в таблице.

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

insert

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

insert(
    keys, values, name=None
)

Ассоциирует keys со values.

Аргументы
keys Ключи для вставки. Может быть тензором любой формы. Должен соответствовать типу ключа таблицы.
values Значения, которые будут ассоциированы с ключами. Должен быть тензором той же формы, что и keys, и соответствовать типу значения таблицы.
name Имя операции (необязательно).
Возвращаемое значение
Созданная операция.
Исключения
TypeError когда keys или values не соответствуют типам данных таблицы.

lookup

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

lookup(
    keys, dynamic_default_values=None, name=None
)

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

default_value используется для ключей, отсутствующих в таблице.

Аргументы
keys Ключи для поиска. Может быть тензором любой формы. Должен соответствовать типу key_dtype таблицы.
dynamic_default_values Значения по умолчанию для отсутствующих ключей в таблице. Если None (по умолчанию), используется значение по умолчанию. Форма dynamic_default_values должна совпадать с формой table.default_value или тензором результата поиска. В последнем случае для каждого ключа будет использоваться другое значение по умолчанию.

Например:

keys = [0, 1, 3]
dynamic_default_values = [[1, 3, 4], [2, 3, 9], [8, 3, 0]]

# The key '0' will use [1, 3, 4] as default value.
# The key '1' will use [2, 3, 9] as default value.
# The key '3' will use [8, 3, 0] as default value.
name Имя операции (необязательно).
Возвращаемое значение
Тензор, содержащий значения той же формы, что и keys, используя тип значения таблицы.
Исключения
TypeError когда keys не соответствуют типам данных таблицы.

remove

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

remove(
    keys, name=None
)

Удаляет keys и его связанные значения из таблицы.

Если ключ отсутствует в таблице, он игнорируется.

Аргументы
keys Ключи для удаления. Может быть тензором любой формы. Должен соответствовать типу ключа таблицы.
name Имя операции (необязательно).
Возвращаемое значение
Созданная операция.
Исключения
TypeError когда keys не соответствуют типам данных таблицы.

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/experimental/MutableHashTable

Spec-Zone.ru

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