tf.raw_ops.EnqueueTPUEmbeddingSparseBatch

Операция, которая добавляет в очередь индексы входных данных TPUEmbedding из SparseTensor.

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

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

См. Руководство по миграции для получения дополнительных сведений.

tf.compat.v1.raw_ops.EnqueueTPUEmbeddingSparseBatch

tf.raw_ops.EnqueueTPUEmbeddingSparseBatch(
    sample_indices, embedding_indices, aggregation_weights, mode_override,
    device_ordinal=-1, combiners=[], name=None
)

Эта операция облегчает перенос кода, использующего embedding_lookup_sparse(), хотя для создания аргументов этой операции требуется предварительная обработка SparseTensor в Python, так как на один шаг обучения разрешено только одна операция EnqueueTPUEmbeddingSparseBatch.

Тензоры в соответствующих позициях в трёх списках входных данных должны иметь одинаковую форму, т.е. ранг 1 с dim_size(), равным общему количеству обращений к таблице, описанной соответствующим table_id.

Аргументы
`sample_indices`	Список из как минимум 1 `Tensor` объекта одного типа в: `int32`, `int64`. Список тензоров ранга 1, определяющих пример обучения и признак, к которому относятся соответствующие значения embedding_indices и aggregation_weights. sample_indices[i] должно быть равно b * nf + f, где nf — количество признаков из соответствующей таблицы, f находится в диапазоне [0, nf), а b находится в диапазоне [0, размер пакета).
`embedding_indices`	Список той же длины, что и `sample_indices`, `Tensor` объектов одного типа в: `int32`, `int64`. Список тензоров ранга 1, индексы в таблицах встраивания.
`aggregation_weights`	Список той же длины, что и `sample_indices`, `Tensor` объектов одного типа в: `float32`, `float64`. Список тензоров ранга 1, содержащих весовые коэффициенты агрегирования для каждого примера — т.е. для каждого (примера обучения, признака).
`mode_override`	`Tensor` типа `string`. Строковый вход, который переопределяет режим, указанный в TPUEmbeddingConfiguration. Поддерживаемые значения: {'unspecified', 'inference', 'training', 'backward_pass_only'}. При значении 'unspecified' используется режим, установленный в TPUEmbeddingConfiguration, в противном случае используется mode_override.
`device_ordinal`	Необязательный `int`. По умолчанию `-1`. TPU-устройство для использования. Должно быть >= 0 и меньше количества TPU-ядер в задаче, на которой размещён узел.
`combiners`	Необязательный список `strings`. По умолчанию `[]`. Список строковых скаляров, по одному для каждой таблицы встраивания, который определяет способ нормализации активаций встраивания после взвешенного суммирования. Поддерживаемые комбинаторы: 'mean', 'sum' или 'sqrtn'. Недопустимо, чтобы сумма весов была равна 0 для 'mean' или сумма квадратов весов равнялась 0 для 'sqrtn'. Если combiners не передаётся, по умолчанию используется 'sum' для всех таблиц.
`name`	Имя операции (необязательно).

Возвращаемое значение
Созданная операция.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/raw_ops/EnqueueTPUEmbeddingSparseBatch