общедоступный статический класс EnqueueTPUEmbeddingRaggedTensorBatch.Options
Необязательные атрибуты для EnqueueTPUEmbeddingRaggedTensorBatch
Публичные методы
EnqueueTPUEmbeddingRaggedTensorBatch.Options | объединители (объединители List<String>) |
EnqueueTPUEmbeddingRaggedTensorBatch.Options | deviceOrdinal (длинный номер устройства) |
EnqueueTPUEmbeddingRaggedTensorBatch.Options | maxSequenceLengths (Список<Long> maxSequenceLengths) |
Унаследованные методы
Публичные методы
общедоступные объединители EnqueueTPUEmbeddingRaggedTensorBatch.Options (объединители List<String>)
Параметры
объединители | Список строковых скаляров, по одному для каждой таблицы внедрения, которые определяют, как нормализовать активации внедрения после взвешенного суммирования. Поддерживаемые объединители: «среднее», «сумма» или «квадрат». Недопустимо, чтобы сумма весов была равна 0 для «среднего» или сумма квадратов весов была равна 0 для «sqrtn». Если объединители не переданы, по умолчанию для всех таблиц используется «сумма». |
---|
public EnqueueTPUEmbeddingRaggedTensorBatch.Options deviceOrdinal (Long deviceOrdinal)
Параметры
устройствоПорядковый номер | Используемое устройство ТПУ. Должно быть >= 0 и меньше количества ядер TPU в задаче, на которой размещен узел. |
---|