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
batch x features x depth if axis == -1
batch x depth x features if axis == 1
depth x batch x features if axis == 0
Supposons que
indices = [0, 2, -1, 1]
depth = 3
on_value = 5.0
off_value = 0.0
axis = -1
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)
indices = [0, 2, -1, 1]
depth = 3
on_value = 0.0
off_value = 3.0
axis = 0
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)
indices = [[0, 2], [1, -1]]
depth = 3
on_value = 1.0
off_value = 0.0
axis = -1
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 |
Retours
- une nouvelle instance de OneHot