tensoreflusso:: ops:: TopK

#include <nn_ops.h>

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). Così,

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.

Resi:

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

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 pubbliche statiche

Sorted (bool x)

Strutture

tensorflow:: ops:: TopK:: Attrs

Setter di attributi facoltativi 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 pubbliche statiche

Ordinato 

Attrs Sorted(
  bool x
)