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