tensoreflusso:: ops:: TopK

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

Riepilogo

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 k voci in ciascuna riga (rispettivamente vettore lungo l'ultima dimensione). Pertanto,

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

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

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 ciascuna riga per le matrici).

Attributi facoltativi (vedi Attrs ):

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

Ritorna:

• Valori Output : i k elementi più grandi lungo ciascuna ultima fetta 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
)