Renvoie un tenseur one-hot.
Les emplacements représentés par des indices dans `indices` prennent la valeur `on_value`, tandis que tous les autres emplacements prennent la valeur `off_value`.
Si l'entrée `indices` est de rang `N`, la sortie aura le rang `N+1`. Le nouvel axe est créé à la dimension `axis` (par défaut : le nouvel axe est ajouté à la fin).
Si « indices » est un scalaire, la forme de sortie sera un vecteur de longueur « profondeur ».
Si `indices` est un vecteur de longueur `features`, la forme de sortie sera :
features x depth if axis == -1
depth x features if axis == 0
Si `indices` est une matrice (lot) avec la forme `[lot, caractéristiques]`, la forme de sortie sera : batch x features x depth if axis == -1
batch x depth x features if axis == 1
depth x batch x features if axis == 0
Exemples =========Supposons que
indices = [0, 2, -1, 1]
depth = 3
on_value = 5.0
off_value = 0.0
axis = -1
Alors la sortie est `[4 x 3]` : output =
[5.0 0.0 0.0] // one_hot(0)
[0.0 0.0 5.0] // one_hot(2)
[0.0 0.0 0.0] // one_hot(-1)
[0.0 5.0 0.0] // one_hot(1)
Supposons que indices = [0, 2, -1, 1]
depth = 3
on_value = 0.0
off_value = 3.0
axis = 0
Alors la sortie est `[3 x 4]` : output =
[0.0 3.0 3.0 3.0]
[3.0 3.0 3.0 0.0]
[3.0 3.0 3.0 3.0]
[3.0 0.0 3.0 3.0]
// ^ one_hot(0)
// ^ one_hot(2)
// ^ one_hot(-1)
// ^ one_hot(1)
Supposons que indices = [[0, 2], [1, -1]]
depth = 3
on_value = 1.0
off_value = 0.0
axis = -1
Alors la sortie est `[2 x 2 x 3]` : output =
[
[1.0, 0.0, 0.0] // one_hot(0)
[0.0, 0.0, 1.0] // one_hot(2)
][
[0.0, 1.0, 0.0] // one_hot(1)
[0.0, 0.0, 0.0] // one_hot(-1)
]
Classes imbriquées
classe | OneHot.Options | Attributs facultatifs pour OneHot |
Constantes
Chaîne | OP_NAME | Le nom de cette opération, tel que connu par le moteur principal TensorFlow |
Méthodes publiques
Sortie <U> | comme Sortie () Renvoie le handle symbolique du tenseur. |
statique OneHot.Options | axe (axe long) |
statique <U étend TType > OneHot <U> | |
Sortie <U> | sortir () Le tenseur brûlant. |
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
Méthodes publiques
sortie publique <U> asOutput ()
Renvoie le handle symbolique du tenseur.
Les entrées des opérations TensorFlow sont les sorties d'une autre opération TensorFlow. Cette méthode est utilisée pour obtenir un handle symbolique qui représente le calcul de l’entrée.
Axe public statique OneHot.Options (axe long)
Paramètres
axe | L'axe à remplir (par défaut : -1, un nouvel axe le plus intérieur). |
---|
public static OneHot <U> créer (Portée de portée , Opérande <? étend TNumber > indices, Opérande < TInt32 > profondeur, Opérande <U> onValue, Opérande <U> offValue, Options... options)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération OneHot.
Paramètres
portée | portée actuelle |
---|---|
indices | Un tenseur d'indices. |
profondeur | Un scalaire définissant la profondeur de la dimension chaude. |
surValeur | Un scalaire définissant la valeur à remplir dans la sortie lorsque `indices[j] = i`. |
valeur désactivée | Un scalaire définissant la valeur à remplir dans la sortie lorsque `indices[j] != i`. |
choix | porte des valeurs d'attributs facultatifs |
Retour
- une nouvelle instance de OneHot