tf.raw_ops.EnqueueTPUEmbeddingRaggedTensorBatch

Упрощает перенос кода, использующего tf.nn.embedding_lookup().

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

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

Подробнее см. в руководстве по миграции.

tf.compat.v1.raw_ops.EnqueueTPUEmbeddingRaggedTensorBatch

tf.raw_ops.EnqueueTPUEmbeddingRaggedTensorBatch(
    sample_splits,
    embedding_indices,
    aggregation_weights,
    mode_override,
    table_ids,
    device_ordinal=-1,
    combiners=[],
    max_sequence_lengths=[],
    num_features=[],
    name=None
)

sample_splits[i], embedding_indices[i] и aggregation_weights[i] соответствуют i-му признаку. table_ids[i] указывает, в какой таблице встраиваний следует искать i-й признак.

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

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

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

© 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/raw_ops/EnqueueTPUEmbeddingRaggedTensorBatch