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]
- `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.
Metodi pubblici
statico <T estende Numero, U, V estende Numero> RaggedGather <T, U> | |
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
Metodi pubblici
public static RaggedGather <T, U> create ( Scope scope, Iterable< Operand <T>> paramsNestedSplits, Operand <U> paramsDenseValues, Operand <V> 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
public List< Output <T>> outputNestedSplits ()
I tensori `nested_row_splits` che definiscono il partizionamento delle righe per il RaggedTensor restituito.