EnqueueTPUEmbeddingBatch

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

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

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

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

сорт EnqueueTPUEmbeddingBatch.Options Дополнительные атрибуты для EnqueueTPUEmbeddingBatch

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

статический EnqueueTPUEmbeddingBatch.Options
объединители (объединители List<String>)
статический EnqueueTPUEmbeddingBatch
create (область области действия , пакет Iterable< Operand <String>>, Operand <String> modeOverride, параметры... )
Фабричный метод для создания класса, обертывающего новую операцию EnqueueTPUEmbeddingBatch.
статический EnqueueTPUEmbeddingBatch.Options
deviceOrdinal (длинный номер устройства)

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

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

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

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

public static EnqueueTPUEmbeddingBatch create (область области действия , пакет Iterable< Operand <String>>, Operand <String> modeOverride, параметры ... )

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

Параметры
объем текущий объем
партия Список одномерных тензоров, по одному для каждой таблицы внедрения, содержащий пакетные входные данные, закодированные как прототипы dist_belief.SparseFeatures. Если поле веса в прототипе SparseFeatures не заполнено для идентификатора, предполагается вес 1,0.
режимOverride Строковый ввод, который переопределяет режим, указанный в TPUEmbeddingConfiguration. Поддерживаемые значения: {'unspecified', 'inference', 'training', 'backward_pass_only'}. Если установлено значение «не указано», используется режим, установленный в TPUEmbeddingConfiguration, в противном случае используется mode_override.
параметры содержит значения необязательных атрибутов
Возврат
  • новый экземпляр EnqueueTPUEmbeddingBatch

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

Параметры
устройствоПорядковый номер Используемое устройство ТПУ. Это значение должно быть -1, когда Op работает на устройстве TPU, и >= 0, когда Op работает на устройстве CPU.