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