tensorflow::ops::TopK

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

Riepilogo

Se l'input è un vettore (di rango 1), trova i k elementi più grandi nel vettore e restituisce i loro valori e indici come vettori. Quindi values[j] è il j -esimo elemento più grande input e il suo indice è indices[j] .

Per le matrici (risp. input di rango superiore), calcola le prime k voci in ogni riga (risp. vettore lungo l'ultima dimensione). Pertanto,

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

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

Argomentazioni:

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

Attributi opzionali (vedere Attrs ):

• ordinato: se impostato su true, i k elementi risultanti saranno ordinati in base ai valori in ordine decrescente.

Resi:

• Valori Output : I k elementi più grandi lungo ciascuna ultima sezione dimensionale.
• Indici Output : Gli indici dei values all'interno dell'ultima dimensione di input .

attributi pubblici

::tensorflow::Output indices

Operation operation

::tensorflow::Output values

 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
)

Attrs Sorted(
  bool x
)