Aprenda o que há de mais recente em aprendizado de máquina, IA generativa e muito mais no WiML Symposium 2023 Registre-se

Esta página foi traduzida pela API Cloud Translation.

RecvTPUEmbeddingActivations

classe final pública RecvTPUEmbeddingActivations

Uma operação que recebe ativações de incorporação na TPU.

O sistema TPU executa as pesquisas de incorporação e agregações especificadas pelos argumentos para TPUEmbeddingEnqueue (Integer / Sparse / SparseTensor) Batch. Os resultados dessas agregações são visíveis no Tensorflow Graph como as saídas de uma operação RecvTPUEmbeddingActivations. Esta operação retorna uma lista contendo um Tensor de ativações por tabela especificada no modelo. Pode haver no máximo uma op RecvTPUEmbeddingActivations no gráfico de TPU.

Métodos Públicos

static RecvTPUEmbeddingActivations	criar ( escopo do escopo, numOutputs longos, configuração de string) Método de fábrica para criar uma classe que envolve uma nova operação RecvTPUEmbeddingActivations.
Iterador < Operando <Float>>	iterador ()
Listar < Output <Float>>	saídas () Um TensorList de ativações de incorporação contendo um Tensor por tabela de incorporação no modelo.

Métodos herdados

Da classe org.tensorflow.op.PrimitiveOp

final booleano	igual a (objeto obj)
final int	hashCode ()
Operação	op () Retorna a `Operation` subjacente
corda final	toString ()

Da classe java.lang.Object

boleano	igual a (objeto arg0)
aula final <?>	getClass ()
int	hashCode ()
vazio final	notificar ()
vazio final	notificar tudo ()
Fragmento	toString ()
vazio final	espera (long arg0, int arg1)
vazio final	espera (long arg0)
vazio final	espera ()

Da interface java.lang.Iterable

vazio abstrato	forEach (Consumidor <? super T> arg0)
Iterador abstrato < Operando <Float>>	iterador ()
Divisor abstrato < Operando <Float>>	divisor ()

Métodos Públicos

public static RecvTPUEmbeddingActivations create ( escopo de escopo, Long numOutputs, configuração de string)

Método de fábrica para criar uma classe que envolve uma nova operação RecvTPUEmbeddingActivations.

Parâmetros

alcance	escopo atual
numOutputs	O número de tensores de ativação de saída, igual ao número de tabelas de incorporação no modelo.
config	Protocolo TPUEmbeddingConfiguration serializado.

Devoluções

uma nova instância de RecvTPUEmbeddingActivations

Iterador público < Operando <Float>> iterador ()

public List < Output <Float>> outputs ()

Um TensorList de ativações de incorporação contendo um Tensor por tabela de incorporação no modelo.