classe final pública XlaRecvTPUEmbeddingActivations
Uma operação que recebe ativações de incorporação na TPU.
O sistema TPU realiza pesquisas e agregações de incorporação. Os resultados dessas agregações são visíveis para o Tensorflow Graph como as saídas de uma operação XlaRecvTPUEmbeddingActivations. Esta operação retorna uma lista contendo um Tensor de ativações por tabela especificada no modelo.
Métodos Públicos
| XlaRecvTPUEmbeddingActivations estáticas | |
| Iterador< Operando <Float>> | iterador () |
| Lista< Saída <Float>> | saídas () Um TensorList de ativações de incorporação contendo um Tensor por tabela de incorporação no modelo. |
Métodos herdados
Métodos Públicos
public static XlaRecvTPUEmbeddingActivations create ( Escopo de escopo , Operando <?> deduplicationData, Long numTables, String config)
Método de fábrica para criar uma classe que agrupa uma nova operação XlaRecvTPUEmbeddingActivations.
Parâmetros
| escopo | escopo atual |
|---|---|
| desduplicaçãoDados | Um Tensor com type=DT_VARIANT contendo os dados de desduplicação. O tensor é uma tupla aninhada XLA contendo N elementos (onde N é a razão entre o número de núcleos incorporados e tensores por chip TPU). Cada elemento da tupla aninhada é uma tupla de tensores de classificação 1. Cada tensor contém índices (DT_UINT32) para incorporar a pesquisa no TensorCore ou pesos (DT_FLOAT) para aplicar à saída da operação de pesquisa de incorporação. |
| numTables | O número de tensores de ativação de saída. Se o descritor de recurso estiver presente na configuração de incorporação tpu, ele será igual ao número de recursos, caso contrário, será igual ao número de tabelas de incorporação no modelo. |
| configuração | Protocolo TPUEmbeddingConfiguration serializado. |
Devoluções
- uma nova instância de XlaRecvTPUEmbeddingActivations
Lista pública< Saída <Float>> saídas ()
Um TensorList de ativações de incorporação contendo um Tensor por tabela de incorporação no modelo.