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 |
Méthodes publiques
Sortie <U> | comme Sortie () Renvoie le handle symbolique d'un tenseur. |
statique OneHot.Options | axe (axe long) |
statique <U, T étend le numéro> OneHot <U> | |
Sortie <U> | sortir () Le tenseur brûlant. |
Méthodes héritées
Méthodes publiques
sortie publique <U> asOutput ()
Renvoie le handle symbolique d'un 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 la portée, indices de l'opérande <T>, profondeur de l'opérande <Entier>, 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