публичный финальный класс 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 | |
Публичные методы
| статический EnqueueTPUEmbeddingRaggedTensorBatch.Options | объединители (объединители List<String>) |
| static <T расширяет число, U расширяет число, V расширяет число> EnqueueTPUEmbeddingRaggedTensorBatch | create ( Область видимости , Iterable< Operand <T>> sampleSplits, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand <String> modeOverride, List<Long> tableIds, Options... options) Фабричный метод для создания класса, обертывающего новую операцию EnqueueTPUEmbeddingRaggedTensorBatch. |
| статический EnqueueTPUEmbeddingRaggedTensorBatch.Options | deviceOrdinal (длинный номер устройства) |
| статический EnqueueTPUEmbeddingRaggedTensorBatch.Options | maxSequenceLengths (Список<Long> maxSequenceLengths) |
| статический EnqueueTPUEmbeddingRaggedTensorBatch.Options | numFeatures (Список<Long> numFeatures) |
Унаследованные методы
Публичные методы
общедоступные статические объединители EnqueueTPUEmbeddingRaggedTensorBatch.Options (объединители List<String>)
Параметры
| объединители | Список строковых скаляров, по одному для каждой таблицы внедрения, которые определяют, как нормализовать активации внедрения после взвешенного суммирования. Поддерживаемые объединители: «среднее», «сумма» или «квадрат». Недопустимо, чтобы сумма весов была равна 0 для «среднего» или сумма квадратов весов была равна 0 для «sqrtn». Если объединители не переданы, по умолчанию для всех таблиц используется «сумма». |
|---|
public static EnqueueTPUEmbeddingRaggedTensorBatch create ( Область действия, Iterable< Operand <T>> sampleSplits, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand <String> 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 в задаче, на которой размещен узел. |
|---|