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.