Recebe dados de desduplicação (índices e pesos) do núcleo de incorporação.
Os dados de desduplicação são um tensor com type=DT_VARIANT. O próprio tensor é uma tupla aninhada XLA contendo N elementos (onde N é a razão do número de incorporação de 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.
Métodos públicos
Saída <Objeto> | comoSaída () Retorna o identificador simbólico de um tensor. |
static XlaRecvTPUEembeddingDeduplicationData | |
Saída <?> | saída () |
Métodos Herdados
Métodos públicos
Public Output <Object> asOutput ()
Retorna o identificador simbólico de um tensor.
As entradas para operações do TensorFlow são saídas de outra operação do TensorFlow. Este método é usado para obter um identificador simbólico que representa o cálculo da entrada.
public static XlaRecvTPUEmbeddingDeduplicationData create ( escopo de escopo, configuração de string)
Método de fábrica para criar uma classe que envolve uma nova operação XlaRecvTPUEmbeddingDeduplicationData.
Parâmetros
alcance | escopo atual |
---|---|
configuração | Protocolo serializado de configuração TPUEmbedding. |
Devoluções
- uma nova instância de XlaRecvTPUEmbeddingDeduplicationData