публичный финальный класс RecvTPUEmbeddingActivations
Операция, которая получает активации встраивания в TPU.
Система TPU выполняет поиск по внедрению и агрегацию, заданную аргументами TPUEmbeddingEnqueue(Integer/Sparse/SparseTensor)Batch. Результаты этих агрегаций видны в Tensorflow Graph как выходные данные операции RecvTPUEmbeddingActivations. Эта операция возвращает список, содержащий один тензор активаций для каждой таблицы, указанной в модели. В графе TPU может быть не более одной операции RecvTPUEmbeddingActivations.
Константы
| Нить | OP_NAME | Название этой операции, известное основному движку TensorFlow. |
Публичные методы
| статические RecvTPUEmbeddingActivations | create (область области действия , длинное количество выходов, строковая конфигурация) Фабричный метод для создания класса, обертывающего новую операцию RecvTPUEmbeddingActivations. |
| Итератор< Операнд <TFloat32> > | итератор () |
| Список < Вывод < TFloat32 >> | выходы () TensorList активаций внедрения, содержащий по одному тензору на каждую таблицу внедрения в модели. |
Унаследованные методы
Константы
общедоступная статическая финальная строка OP_NAME
Название этой операции, известное основному движку TensorFlow.
Постоянное значение: «RecvTPUEmbeddingActivations».
Публичные методы
public static RecvTPUEmbeddingActivations create (область области , Long numOutputs, строковая конфигурация)
Фабричный метод для создания класса, обертывающего новую операцию RecvTPUEmbeddingActivations.
Параметры
| объем | текущий объем |
|---|---|
| числовыходов | Количество выходных тензоров активации, равное количеству таблиц встраивания в модель. |
| конфигурация | Сериализованный прототип TPUEmbeddingConfiguration. |
Возврат
- новый экземпляр RecvTPUEmbeddingActivations
общедоступный список< Вывод <TFloat32> > выходы ()
TensorList активаций внедрения, содержащий по одному тензору на каждую таблицу внедрения в модели.