RaggedGather

classe finale pubblica RaggedGather

Raccogli le sezioni irregolari dall'asse "params" "0" in base agli "indici".

Genera un output `RaggedTensor` composto da `output_dense_values` e `output_nested_splits`, in modo tale che:

output.shape = indices.shape + params.shape[1:]
 output.ragged_rank = indices.shape.ndims + params.ragged_rank
 output[i...j, d0...dn] = params[indices[i...j], d0...dn]
Dove
  • `params = ragged.from_nested_row_splits(params_dense_values, params_nested_splits)` fornisce i valori che dovrebbero essere raccolti.
  • Gli "indici" sono un tensore denso con dtype "int32" o "int64", che indica quali valori devono essere raccolti.
  • "output = ragged.from_nested_row_splits(output_dense_values, output_nested_splits)" è il tensore di output.
(Nota: questa operazione C++ viene utilizzata per implementare l'operazione Python di livello superiore `tf.ragged.gather`, che supporta anche indici irregolari.)

Costanti

Corda OP_NAME Il nome di questa operazione, come noto al motore principale di TensorFlow

Metodi pubblici

statico <T estende TNumber , U estende TType > RaggedGather <T, U>
create ( Scope scope, Iterable< Operand <T>> paramsNestedSplits, Operand <U> paramsDenseValues, Operand <? extends TNumber > indicis, Long OUTPUTRAGGEDRANK)
Metodo factory per creare una classe che racchiude una nuova operazione RaggedGather.
Uscita <U>
outputDenseValues ​​()
I `flat_values` per il RaggedTensor restituito.
Elenco< Uscita <T>>
outputNestedSplits ()
I tensori `nested_row_splits` che definiscono il partizionamento delle righe per il RaggedTensor restituito.

Metodi ereditati

org.tensorflow.op.RawOp
booleano finale
è uguale a (Oggetto oggetto)
finale int
codicehash ()
Operazione
op ()
Restituisce questa unità di calcolo come una singola Operation .
stringa finale
aStringa ()
booleano
è uguale a (Oggetto arg0)
Classe finale<?>
getClass ()
int
codicehash ()
vuoto finale
avvisare ()
vuoto finale
notificaTutti ()
Corda
aStringa ()
vuoto finale
attendere (lungo arg0, int arg1)
vuoto finale
aspetta (lungo arg0)
vuoto finale
Aspettare ()
org.tensorflow.op.Op
ambiente di esecuzione astratto
ambiente ()
Restituisce l'ambiente di esecuzione in cui è stata creata questa operazione.
Operazione astratta
op ()
Restituisce questa unità di calcolo come una singola Operation .

Costanti

Stringa finale statica pubblica OP_NAME

Il nome di questa operazione, come noto al motore principale di TensorFlow

Valore costante: "RaggedGather"

Metodi pubblici

public static RaggedGather <T, U> create ( Scope scope, Iterable< Operand <T>> paramsNestedSplits, Operand <U> paramsDenseValues, Operand <? extends TNumber > indici, Long OUTPUTRAGGEDRANK)

Metodo factory per creare una classe che racchiude una nuova operazione RaggedGather.

Parametri
ambito ambito attuale
paramsNestedSplits I tensori `nested_row_splits` che definiscono il partizionamento delle righe per l'input RaggedTensor `params`.
paramsDenseValues I "flat_values" per il RaggedTensor "params". C'è stata una modifica alla terminologia a livello di Python da dense_values ​​a flat_values, quindi dense_values ​​è il nome deprecato.
indici Indici nella dimensione più esterna di "params" dei valori che dovrebbero essere raccolti.
OUTPUTRAGGEDRANK Il rango irregolare dell'output RaggedTensor. "output_nested_splits" conterrà questo numero di tensori "row_splits". Questo valore dovrebbe essere uguale a "indices.shape.ndims + params.ragged_rank - 1".
Ritorni
  • una nuova istanza di RaggedGather

output pubblico <U> outputDenseValues ​​()

I `flat_values` per il RaggedTensor restituito.

public List< Output <T>> outputNestedSplits ()

I tensori `nested_row_splits` che definiscono il partizionamento delle righe per il RaggedTensor restituito.