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 |
Costanti
Corda | OP_NAME | Il nome di questa operazione, come noto al motore principale di TensorFlow |
Metodi pubblici
Uscita <T> | comeuscita () Restituisce l'handle simbolico del tensore. |
Gather.Options statico | batchDims (batchDims lungo) |
statico <T estende TType > Raccogli <T> | |
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
Costanti
Stringa finale statica pubblica OP_NAME
Il nome di questa operazione, come noto al motore principale di TensorFlow
Metodi pubblici
Uscita pubblica <T> asOutput ()
Restituisce l'handle simbolico del 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 Gather <T> create ( ambito ambito , parametri operando <T>, operando <? estende TNumber > indici, operando <? estende TNumber > asse, 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:]`.