Получает данные дедупликации (индексы и веса) от ядра внедрения.
Данные дедупликации представляют собой тензор с типом = DT_VARIANT. Сам тензор представляет собой вложенный кортеж XLA, содержащий N элементов (где N — отношение количества встраиваемых ядер к тензорным ядрам на чип TPU). Каждый элемент вложенного кортежа представляет собой кортеж тензоров ранга 1. Каждый тензор либо содержит индексы (DT_UINT32) для внедрения поиска в TensorCore, либо веса (DT_FLOAT), которые применяются к выходным данным операции внедрения поиска.
Публичные методы
| Вывод <Объект> | какВывод () Возвращает символический дескриптор тензора. |
| статический XlaRecvTPUEmbeddingDeduplicationData | create (область области действия , конфигурация строки) Фабричный метод для создания класса, обертывающего новую операцию XlaRecvTPUEmbeddingDeduplicationData. |
| Вывод <?> | выход () |
Унаследованные методы
Публичные методы
общедоступный вывод <Object> asOutput ()
Возвращает символический дескриптор тензора.
Входные данные для операций TensorFlow являются выходными данными другой операции TensorFlow. Этот метод используется для получения символического дескриптора, который представляет собой вычисление входных данных.
public static XlaRecvTPUEmbeddingDeduplicationData create (область области , строковая конфигурация)
Фабричный метод для создания класса, обертывающего новую операцию XlaRecvTPUEmbeddingDeduplicationData.
Параметры
| объем | текущий объем |
|---|---|
| конфигурация | Сериализованный прототип TPUEmbeddingConfiguration. |
Возврат
- новый экземпляр XlaRecvTPUEmbeddingDeduplicationData