Um op que recebe ativações de incorporação na TPU.
O sistema TPU executa as 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 Op XlaRecvTPUEmbeddingActivations. Esta op retorna uma lista contendo um Tensor de ativações por tabela especificada no modelo.
Métodos públicos
static XlaRecvTPUEmbeddingAtivações | |
Iterador< Operando <Float>> | iterador () |
Lista< Saída <Float>> | saídas () Uma 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 do escopo, Operando <?> deduplicationData, Long numTables, String config)
Método de fábrica para criar uma classe envolvendo uma nova operação XlaRecvTPUEmbeddingActivations.
Parâmetros
alcance | escopo atual |
---|---|
dados de desduplicação | 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 do número de incorporação para núcleos de tensor por chip TPU). Cada elemento da tupla aninhada é uma tupla de tensores de nível 1. Cada tensor contém índices (DT_UINT32) para pesquisa de incorporação 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 de tpu, ele será igual ao número de recursos, caso contrário, igual ao número de tabelas de incorporação no modelo. |
configuração | Protocolo serializado de configuração TPUEmbedding. |
Devoluções
- uma nova instância de XlaRecvTPUEmbeddingActivations
public List< Output <Float>> outputs ()
Uma TensorList de ativações de incorporação contendo um Tensor por tabela de incorporação no modelo.