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