публичный финальный класс 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. |
---|