Raccogliere

public final class Gather

Raccogli le sezioni dall'asse "params" "asse" in base a "indici".

"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]
 

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

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

Classi annidate

classe Raccogli.Opzioni Attributi opzionali perGather

Metodi pubblici

Uscita <T>
asOutput ()
Restituisce la maniglia simbolica di un tensore.
statico Gather.Options
static <T, U estende il numero, V estende il numero> Gather <T>
create ( ambito ambito, parametri operando <T>, indici operando <U>, asse operando <V>, opzioni ... )
Metodo Factory per creare una classe che avvolge una nuova operazione Gather.
Uscita <T>
uscita ()
Valori da `params` raccolti da indici forniti da` indices`, con forma `params.shape [: axis] + indices.shape + params.shape [axis + 1:]`.

Metodi ereditati

Metodi pubblici

output pubblico <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.

public static Gather.Options batchDims (Long batchDims)

public static Gather <T> create ( Scope scope, Operand <T> params, Operand <U> index, Operand <V> axis, Options ... options)

Metodo Factory per creare una classe che esegue il wrapping di una nuova operazione Gather.

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

output pubblico <T> output ()

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