TPUEmbeddingActivations

public final class TPUEmbeddingActivations

Un'opzione che consente la differenziazione degli incorporamenti TPU.

Questa operazione restituisce semplicemente il suo primo input, che si presume sia stato tagliato dai Tensors restituiti da TPUEmbeddingDequeueActivations. La presenza di questa operazione, e il suo primo argomento è una variabile addestrabile, consente la differenziazione automatica dei grafici contenenti incorporamenti tramite le librerie TPU Embedding Python.

Metodi pubblici

Uscita <Float>
asOutput ()
Restituisce la maniglia simbolica di un tensore.
TPUEmbeddingActivations statiche
create ( ambito ambito, operando <Float> embeddingVariable, operando <Float> slicedActivations, long tableId, long lookupId)
Metodo Factory per creare una classe che avvolge una nuova operazione TPUEmbeddingActivations.
Uscita <Float>
uscita ()

Metodi ereditati

Metodi pubblici

output pubblico <Float> asOutput ()

Restituisce la maniglia simbolica di un tensore.

Gli input per le operazioni TensorFlow sono output di un'altra operazione TensorFlow. Questo metodo viene utilizzato per ottenere un handle simbolico che rappresenta il calcolo dell'input.

public static TPUEmbeddingActivations create ( Scope scope, Operand <Float> embeddingVariable, Operand <Float> slicedActivations, Long tableId, Long lookupId)

Metodo Factory per creare una classe che avvolge una nuova operazione TPUEmbeddingActivations.

Parametri
scopo ambito attuale
embeddingVariable Una variabile addestrabile, che consente agli ottimizzatori di trovare questa operazione.
slicedActivations L'attivazione dell'incorporamento Tensor to return.
tableId L'id della tabella nella configurazione del livello di incorporamento da cui sono state calcolate queste attivazioni.
lookupId Identificatore del set di indici di incorporamento che ha prodotto queste attivazioni.
ritorna
  • una nuova istanza di TPUEmbeddingActivations

output pubblico <Float> output ()