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