TopK

classe finale pubblica TopK

Trova valori e indici degli elementi più grandi `k` per l'ultima dimensione.

Se l'input è un vettore (rango 1), trova le `k` voci più grandi nel vettore e restituisce i relativi valori e indici come vettori. Pertanto "values[j]" è la "j"-esima voce più grande in "input" e il suo indice è "indices[j]".

Per le matrici (rispettivamente input di rango più elevato), calcola le prime voci `k` in ciascuna riga (rispettivamente vettore lungo l'ultima dimensione). Così,

valori.forma = indici.forma = input.forma[:-1] + [k]

Se due elementi sono uguali, viene visualizzato per primo l'elemento con l'indice inferiore.

Classi nidificate

classe TopK.Opzioni Attributi facoltativi per TopK

Costanti

Corda OP_NAME Il nome di questa operazione, come noto al motore principale di TensorFlow

Metodi pubblici

statico <T estende TNumero > TopK <T>
create ( ambito ambito , operando <T> input, operando < TInt32 > k, Opzioni... opzioni)
Metodo factory per creare una classe che racchiude una nuova operazione TopK.
Uscita < TInt32 >
indici ()
Gli indici di "valori" all'interno dell'ultima dimensione di "input".
TopK.Options statico
ordinato (ordinato booleano)
Uscita <T>
valori ()
Gli elementi più grandi "k" lungo ciascuna ultima fetta dimensionale.

Metodi ereditati

Costanti

Stringa finale statica pubblica OP_NAME

Il nome di questa operazione, come noto al motore principale di TensorFlow

Valore costante: "TopKV2"

Metodi pubblici

public static TopK <T> create ( ambito ambito , operando <T> input, operando < TInt32 > k, Opzioni... opzioni)

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

Parametri
scopo ambito attuale
ingresso 1-D o superiore con l'ultima dimensione almeno "k".
K 0-D. Numero di elementi superiori da cercare lungo l'ultima dimensione (lungo ciascuna riga per le matrici).
opzioni trasporta valori di attributi opzionali
ritorna
  • una nuova istanza di TopK

output pubblico < TInt32 > indici ()

Gli indici di "valori" all'interno dell'ultima dimensione di "input".

TopK.Options statiche pubbliche ordinate (ordinate in modo booleano)

Parametri
smistato Se vero, gli elementi "k" risultanti verranno ordinati in base ai valori in ordine decrescente.

valori di output <T> pubblici ()

Gli elementi più grandi "k" lungo ciascuna ultima fetta dimensionale.