EnqueueTPUEmbeddingRaggedTensorBatch

публичный финальный класс EnqueueTPUEmbeddingRaggedTensorBatch

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

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

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

Вложенные классы

сорт EnqueueTPUEmbeddingRaggedTensorBatch.Options Необязательные атрибуты для EnqueueTPUEmbeddingRaggedTensorBatch

Константы

Нить OP_NAME Название этой операции, известное основному движку TensorFlow.

Публичные методы

статический EnqueueTPUEmbeddingRaggedTensorBatch.Options
объединители (объединители List<String>)
static <T расширяет TNumber , U расширяет TNumber , V расширяет TNumber > EnqueueTPUEmbeddingRaggedTensorBatch
create ( Область видимости, Iterable< Operand <T>> sampleSplits, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand < TString > modeOverride, List<Long> tableIds, Options... options)
Фабричный метод для создания класса, обертывающего новую операцию EnqueueTPUEmbeddingRaggedTensorBatch.
статический EnqueueTPUEmbeddingRaggedTensorBatch.Options
deviceOrdinal (длинный номер устройства)
статический EnqueueTPUEmbeddingRaggedTensorBatch.Options
maxSequenceLengths (Список<Long> maxSequenceLengths)

Унаследованные методы

Константы

общедоступная статическая финальная строка OP_NAME

Название этой операции, известное основному движку TensorFlow.

Постоянное значение: «EnqueueTPUEmbeddingRaggedTensorBatch»

Публичные методы

общедоступные статические объединители EnqueueTPUEmbeddingRaggedTensorBatch.Options (объединители List<String>)

Параметры
объединители Список строковых скаляров, по одному для каждой таблицы внедрения, которые определяют, как нормализовать активации внедрения после взвешенного суммирования. Поддерживаемые объединители: «среднее», «сумма» или «квадрат». Недопустимо, чтобы сумма весов была равна 0 для «среднего» или сумма квадратов весов была равна 0 для «sqrtn». Если объединители не переданы, по умолчанию для всех таблиц используется «сумма».

public static EnqueueTPUEmbeddingRaggedTensorBatch create ( Область видимости , Iterable< Operand <T>> sampleSplits, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand < TString > modeOverride, List<Long> tableIds, Options.. . параметры)

Фабричный метод для создания класса, обертывающего новую операцию EnqueueTPUEmbeddingRaggedTensorBatch.

Параметры
объем текущий объем
образецSplits Список тензоров ранга 1, определяющий точки останова для разделения embedding_indices и aggregation_weights на строки. Он соответствует ids.row_splits в embedding_lookup(), когда ids является RaggedTensor.
встраиваниеИндексы Список тензоров ранга 1, индексов в таблицах встраивания. Он соответствует ids.values ​​в embedding_lookup(), когда ids является RaggedTensor.
агрегированиеВес Список тензоров ранга 1, содержащий агрегированные веса для каждого обучающего примера. Оно соответствует полю значений RaggedTensor с теми же строками row_split, что и id в embedding_lookup(), когда ids является RaggedTensor.
режимOverride Строковый ввод, который переопределяет режим, указанный в TPUEmbeddingConfiguration. Поддерживаемые значения: {'unspecified', 'inference', 'training', 'backward_pass_only'}. Если установлено значение «не указано», используется режим, установленный в TPUEmbeddingConfiguration, в противном случае используется mode_override.
идентификаторы таблиц Список целых чисел, определяющих идентификатор таблицы внедрения (смещение TableDescriptor в TPUEmbeddingConfiguration) для поиска соответствующего ввода. i-й вход ищется с помощью table_ids[i]. Размер списка table_ids должен быть равен размеру sample_indices, embedding_indices и aggregation_weights.
параметры содержит значения необязательных атрибутов
Возврат
  • новый экземпляр EnqueueTPUEmbeddingRaggedTensorBatch

общедоступный статический EnqueueTPUEmbeddingRaggedTensorBatch.Options deviceOrdinal (Long deviceOrdinal)

Параметры
устройствоПорядковый номер Используемое устройство ТПУ. Должно быть >= 0 и меньше количества ядер TPU в задаче, на которой размещен узел.

общедоступный статический EnqueueTPUEmbeddingRaggedTensorBatch.Options maxSequenceLengths (List<Long> maxSequenceLengths)