Merken Sie den Termin vor! Google I / O kehrt vom 18. bis 20. Mai zurück Registrieren Sie sich jetzt
Diese Seite wurde von der Cloud Translation API übersetzt.
Switch to English

Tensorflow :: ops :: TopK

#include <nn_ops.h>

Findet Werte und Indizes der k größten Elemente für die letzte Dimension.

Zusammenfassung

Wenn die Eingabe ein Vektor ist (Rang 1), werden die k größten Einträge im Vektor gefunden und ihre Werte und Indizes als Vektoren ausgegeben. Somit ist der values[j] der j te größte Eintrag in der input , und sein Index ist der Index indices[j] .

Berechnet für Matrizen (bzw. Eingabe mit höherem Rang) die oberen k Einträge in jeder Zeile (bzw. Vektor entlang der letzten Dimension). Also

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

Wenn zwei Elemente gleich sind, wird zuerst das Element mit dem niedrigeren Index angezeigt.

Argumente:

  • scope: Ein Scope- Objekt
  • Eingabe: 1-D oder höher mit der letzten Dimension mindestens k .
  • k: 0-D. Anzahl der obersten Elemente, nach denen entlang der letzten Dimension gesucht werden soll (entlang jeder Zeile für Matrizen).

Optionale Attribute (siehe Attrs ):

  • sortiert: Wenn true, werden die resultierenden k Elemente nach den Werten in absteigender Reihenfolge sortiert.

Kehrt zurück:

  • Output : Die k größten Elemente entlang jeder letzten dimensionalen Schicht.
  • Output - Indizes: Die Indizes der values innerhalb der letzten Dimension der input .

Konstruktoren und Destruktoren

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)

Öffentliche Attribute

indices
operation
values

Öffentliche statische Funktionen

Sorted (bool x)

Strukturen

tensorflow :: ops :: TopK :: Attrs

Optionale Attributsetzer für TopK .

Öffentliche Attribute

Indizes

::tensorflow::Output indices

Operation

Operation operation

Werte

::tensorflow::Output values

Öffentliche Funktionen

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
)

Öffentliche statische Funktionen

Sortiert

Attrs Sorted(
  bool x
)