Facilita il porting del codice che utilizza tf.nn.embedding_lookup_sparse().
embedding_indices[i] e aggregation_weights[i] corrispondono alla i-esima caratteristica.
I tensori nelle posizioni corrispondenti nei tre elenchi di input (sample_indices, embedding_indices e aggregation_weights) devono avere la stessa forma, cioè rango 1 con dim_size() uguale al numero totale di ricerche nella tabella descritta dalla funzione corrispondente.
Classi nidificate
| classe | EnqueueTPUEmbeddingArbitraryTensorBatch.Options | Attributi facoltativi per EnqueueTPUEmbeddingArbitraryTensorBatch | |
Metodi pubblici
| static EnqueueTPUEmbeddingArbitraryTensorBatch.Options | combinatori (combinatori List<String>) |
| static <T estende il numero, U estende il numero, V estende il numero> EnqueueTPUEmbeddingArbitraryTensorBatch | create ( Scope scope, Iterable< Operando <T>> sampleIndicesOrRowSplits, Iterable< Operando <U>> embeddingIndices, Iterable< Operando <V>> aggregationWeights, Operando <String> modeOverride, Opzioni... opzioni) Metodo factory per creare una classe che racchiude una nuova operazione EnqueueTPUEmbeddingArbitraryTensorBatch. |
| static EnqueueTPUEmbeddingArbitraryTensorBatch.Options | dispositivoOrdinale (Dispositivo lungoOrdinale) |
Metodi ereditati
Metodi pubblici
combinatori pubblici statici EnqueueTPUEmbeddingArbitraryTensorBatch.Options (combinatori List<String>)
Parametri
| combinatori | Un elenco di scalari di stringa, uno per ciascuna tabella di incorporamento che specifica come normalizzare le attivazioni di incorporamento dopo la somma ponderata. I combinatori supportati sono "mean", "sum" o "sqrtn". Non è valido che la somma dei pesi sia 0 per "media" o che la somma dei pesi al quadrato sia 0 per "sqrtn". Se non viene passato il combinatore, l'impostazione predefinita è utilizzare 'sum' per tutte le tabelle. |
|---|
public static EnqueueTPUEmbeddingArbitraryTensorBatch create ( Scope scope, Iterable< Operand <T>> sampleIndicesOrRowSplits, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand <String> modeOverride, Options... options)
Metodo factory per creare una classe che racchiude una nuova operazione EnqueueTPUEmbeddingArbitraryTensorBatch.
Parametri
| scopo | ambito attuale |
|---|---|
| sampleIndicesOrRowSplits | Un elenco di tensori di rango 2 che specificano l'esempio di training a cui appartengono i valori embedding_indices e aggregation_weights corrispondenti. Se la dimensione della sua prima dimensione è 0, assumiamo che ciascun embedding_indices appartenga a un campione diverso. Sono consentiti sia int32 che int64 e verranno convertiti internamente in int32. Oppure un elenco di tensori di rango 1 che specificano le suddivisioni delle righe per suddividere embedding_indices e aggregation_weights in righe. Corrisponde a ids.row_splits in embedding_lookup(), quando ids è un RaggedTensor. Quando si accoda il tensore irregolare ND, solo l'ultima dimensione può essere irregolare. la divisione delle righe è un tensore denso 1-D. Quando è vuoto, presupponiamo che venga passato un tensore denso all'operazione. Sono consentiti sia int32 che int64 e verranno convertiti internamente in int32. |
| embeddingIndices | Un elenco di tensori di rango 1, indici nelle tabelle di incorporamento. Sono consentiti sia int32 che int64 e verranno convertiti internamente in int32. |
| aggregationWeights | Un elenco di tensori di rango 1 contenenti pesi di aggregazione per esempio di training. Sono consentiti sia float32 che float64 e verranno convertiti internamente in float32. |
| modeOverride | Un input di tipo stringa che sovrascrive la modalità specificata in TPUEmbeddingConfiguration. I valori supportati sono {'unspecified', 'inference', 'training', 'backward_pass_only'}. Se impostato su "non specificato", viene utilizzata la modalità impostata in TPUEmbeddingConfiguration, altrimenti viene utilizzato mode_override. |
| opzioni | trasporta valori di attributi opzionali |
ritorna
- una nuova istanza di EnqueueTPUEmbeddingArbitraryTensorBatch
public static EnqueueTPUEmbeddingArbitraryTensorBatch.Options deviceOrdinal (Long deviceOrdinal)
Parametri
| dispositivoOrdinale | Il dispositivo TPU da utilizzare. Dovrebbe essere >= 0 e inferiore al numero di core TPU nell'attività su cui è posizionato il nodo. |
|---|