TopK

classe finale publique TopK

Recherche les valeurs et les indices des éléments « k » les plus grands pour la dernière dimension.

Si l'entrée est un vecteur (rang 1), recherche les « k » plus grandes entrées du vecteur et génère leurs valeurs et indices sous forme de vecteurs. Ainsi, `values[j]` est la `j`-ème plus grande entrée dans `input`, et son index est `indices[j]`.

Pour les matrices (resp. entrée de rang supérieur), calcule les premières entrées « k » dans chaque ligne (resp. vecteur le long de la dernière dimension). Ainsi,

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

Si deux éléments sont égaux, l’élément d’indice inférieur apparaît en premier.

Classes imbriquées

classe

TopK.Options

Attributs facultatifs pour TopK

Constantes

Chaîne

OP_NAME

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

Méthodes publiques

statique <T étend TNumber > TopK <T>	créer ( Portée de portée , Entrée Opérande <T>, Opérande < TInt32 > k, Options... options) Méthode d'usine pour créer une classe encapsulant une nouvelle opération TopK.
Sortie < TInt32 >	indices () Les indices des « valeurs » dans la dernière dimension de « entrée ».
TopK.Options statique	trié (trié booléen)
Sortie <T>	valeurs () Les «k» plus grands éléments le long de chaque dernière tranche dimensionnelle.

Méthodes héritées

De la classe org.tensorflow.op.RawOp

booléen final	est égal (Obj objet)
entier final	Code de hachage ()
Opération	op () Renvoie cette unité de calcul sous la forme d'une seule Operation .
Chaîne finale	àChaîne ()

De la classe java.lang.Object

booléen	est égal à (Objet arg0)
Classe finale<?>	obtenirClasse ()
int	Code de hachage ()
vide final	notifier ()
vide final	notifierTous ()
Chaîne	àChaîne ()
vide final	attendre (long arg0, int arg1)
vide final	attendez (long arg0)
vide final	attendez ()

Depuis l'interface org.tensorflow.op.Op

résuméExécutionEnvironnement	env () Renvoie l'environnement d'exécution dans lequel cette opération a été créée.
opération abstraite	op () Renvoie cette unité de calcul sous la forme d'une seule Operation .

Constantes

chaîne finale statique publique OP_NAME

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

Valeur constante : "TopKV2"

Méthodes publiques

public static TopK <T> créer ( Portée de portée , Entrée Opérande <T>, Opérande < TInt32 > k, Options... options)

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

Paramètres

portée	portée actuelle
saisir	1-D ou supérieur avec la dernière dimension au moins « k ».
k	0-D. Nombre d'éléments supérieurs à rechercher le long de la dernière dimension (le long de chaque ligne pour les matrices).
choix	porte des valeurs d'attributs facultatifs

Retour

• une nouvelle instance de TopK

Sortie publique < TInt32 > indices ()

Les indices des « valeurs » dans la dernière dimension de « entrée ».

public static TopK.Options triés (triés booléens)

Paramètres

trié	Si c'est vrai, les éléments « k » résultants seront triés par valeurs par ordre décroissant.

valeurs de sortie publiques <T> ()

Les «k» plus grands éléments le long de chaque dernière tranche dimensionnelle.