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	batchDims ( batchDims lungo)
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

Dalla classe org.tensorflow.op.PrimitiveOp

booleano finale	è uguale a (Object obj)
final int	hashCode ()
Operazione	op () Restituisce l' Operation sottostante
stringa finale	toString ()

Dalla classe java.lang.Object

booleano	è uguale a (Oggetto arg0)
classe finale <?>	getClass ()
int	hashCode ()
vuoto finale	notifica ()
vuoto finale	notifyAll ()
Corda	toString ()
vuoto finale	wait (long arg0, int arg1)
vuoto finale	aspetta (lungo arg0)
vuoto finale	aspetta ()

Dall'interfaccia org.tensorflow.Operand

output astratto <T>

asOutput () Restituisce la maniglia simbolica di un tensore.

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:]`.