TopKUnique

cours final public TopKUnique

Renvoie les valeurs uniques TopK dans le tableau par ordre trié. Le

le temps d'exécution est proportionnel au produit de K et de la taille d'entrée. Le tri de l'ensemble du tableau est plus efficace pour des valeurs de K suffisamment grandes. L'algorithme de médiane des médianes est probablement plus rapide, mais difficile à implémenter efficacement dans XLA. S'il y a moins de K nombres uniques (pas de NAN), les résultats sont complétés avec un infini négatif. Les NaN ne sont jamais retournés. Les nombres anormaux sont ramenés à zéro. Si un élément apparaît à plusieurs indices, l'indice le plus élevé est renvoyé. Si un élément TopK n'apparaît jamais dans l'entrée en raison des valeurs de remplissage, les indices sont complétés par un négatif. Si une valeur de remplissage apparaît dans l'entrée et qu'un remplissage est nécessaire, l'index le plus élevé de la valeur de remplissage sera renvoyé. La sémantique n'est pas la même que celle de kth_order_statistic.

Constantes

Chaîne OP_NAME Le nom de cette opération, tel que connu par le moteur principal TensorFlow

Méthodes publiques

statique TopKUnique
créer ( Portée de portée , Opérande < TFloat32 > entrée, Long k)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération TopKUnique.
Sortie < TFloat32 >
topk ()
Sortie < TInt32 >

Méthodes héritées

Constantes

chaîne finale statique publique OP_NAME

Le nom de cette opération, tel que connu par le moteur principal TensorFlow

Valeur constante : "TopKUnique"

Méthodes publiques

public static TopKUnique create ( Portée , Opérande < TFloat32 > entrée, Long k)

Méthode d'usine pour créer une classe encapsulant une nouvelle opération TopKUnique.

Paramètres
portée portée actuelle
Retour
  • une nouvelle instance de TopKUnique

Sortie publique < TFloat32 > topk ()

Sortie publique < TInt32 > topkIndices ()