Questa pagina è stata tradotta dall'API Cloud Translation.
Switch to English

tensorflow :: ops :: TopK

#include <nn_ops.h>

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

Sommario

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

Per le matrici (risp. Input di rango superiore), calcola le prime k voci in ogni riga (risp. Vettore lungo l'ultima dimensione). Quindi,

values.shape = indices.shape = input.shape[:-1] + [k]

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

Argomenti:

  • scope: un oggetto Scope
  • input: 1-D o superiore con l'ultima dimensione almeno k .
  • k: 0-D. Numero di elementi superiori da cercare lungo l'ultima dimensione (lungo ogni riga per le matrici).

Attributi opzionali (vedi Attrs ):

  • ordinato: se vero, i k elementi risultanti verranno ordinati in base ai valori in ordine decrescente.

Ritorna:

  • Valori di Output : i k elementi più grandi lungo ogni ultima fetta dimensionale.
  • Indici di Output : gli indici dei values all'interno dell'ultima dimensione input .

Costruttori e distruttori

TopK (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input k)
TopK (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input k, const TopK::Attrs & attrs)

Attributi pubblici

indices
operation
values

Funzioni statiche pubbliche

Sorted (bool x)

Structs

tensorflow :: ops :: TopK :: Attrs

Setter attributi opzionali per TopK .

Attributi pubblici

indici

::tensorflow::Output indices

operazione

Operation operation

valori

::tensorflow::Output values

Funzioni pubbliche

TopK

 TopK(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input k
)

TopK

 TopK(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input k,
  const TopK::Attrs & attrs
)

Funzioni statiche pubbliche

smistato

Attrs Sorted(
  bool x
)