Raccogli

Riunione pubblica della lezione finale

Raccogli le sezioni dall'asse `params` `axis` in base a `indices`.

"indici" deve essere un tensore intero di qualsiasi dimensione (solitamente 0-D o ​​1-D). Produce un tensore di output con forma `params.shape[:axis] + indices.shape[batch_dims:] + params.shape[axis + 1:]` dove:

# Scalar indices (output is rank(params) - 1).
     output[a_0, ..., a_n, b_0, ..., b_n] =
       params[a_0, ..., a_n, indices, b_0, ..., b_n]
 
     # Vector indices (output is rank(params)).
     output[a_0, ..., a_n, i, b_0, ..., b_n] =
       params[a_0, ..., a_n, indices[i], b_0, ..., b_n]
 
     # Higher rank indices (output is rank(params) + rank(indices) - 1).
     output[a_0, ..., a_n, i, ..., j, b_0, ... b_n] =
       params[a_0, ..., a_n, indices[i, ..., j], b_0, ..., b_n]
 

Si noti che sulla CPU, se viene trovato un indice fuori limite, viene restituito un errore. Sulla GPU, se viene trovato un indice fuori limite, viene archiviato uno 0 nel valore di output corrispondente.

Vedi anche `tf.batch_gather` e `tf.gather_nd`.

Classi nidificate

classe Raccogli.Opzioni Attributi facoltativi per Gather

Metodi pubblici

Uscita <T>
comeuscita ()
Restituisce la maniglia simbolica di un tensore.
Gather.Options statico
batchDims (batchDims lungo)
statico <T, U estende il numero, V estende il numero> Raccogli <T>
create ( ambito ambito , parametri operando <T>, indici operando <U>, asse operando <V>, Opzioni... opzioni)
Metodo factory per creare una classe che racchiude una nuova operazione Gather.
Uscita <T>
produzione ()
Valori da `params` raccolti dagli indici forniti da `indices`, con forma `params.shape[:axis] + indices.shape + params.shape[axis + 1:]`.

Metodi ereditati

Metodi pubblici

Uscita pubblica <T> 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.

pubblico statico Gather.Options batchDims (batchDims lungo)

public static Gather <T> create ( ambito ambito , parametri operando <T>, indici operando <U>, asse operando <V>, opzioni... opzioni)

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

Parametri
scopo ambito attuale
parametri Il tensore da cui raccogliere valori. Deve avere almeno il rango "asse + 1".
indici Tensore dell'indice. Deve essere compreso nell'intervallo "[0, params.shape[asse])".
asse L'asse in "params" da cui raccogliere gli "indici". Il valore predefinito è la prima dimensione. Supporta indici negativi.
opzioni trasporta valori di attributi opzionali
ritorna
  • una nuova istanza di Gather

Uscita pubblica <T> uscita ()

Valori da `params` raccolti dagli indici forniti da `indices`, con forma `params.shape[:axis] + indices.shape + params.shape[axis + 1:]`.