Devuelve un tensor one-hot.
Las ubicaciones representadas por índices en "índices" toman el valor "on_value", mientras que todas las demás ubicaciones toman el valor "off_value".
Si la entrada `índices` es rango` N`, la salida tendrá rango `N + 1`. El nuevo eje se crea en la dimensión` eje` (predeterminado: el nuevo eje se agrega al final).
Si "índices" es un escalar, la forma de salida será un vector de longitud "profundidad".
Si `indices` es un vector de longitud` features`, la forma de salida será:
features x depth if axis == -1
depth x features if axis == 0
Si `indices` es una matriz (batch) con forma de` [lote, cuenta] `, la forma de salida será: batch x features x depth if axis == -1
batch x depth x features if axis == 1
depth x batch x features if axis == 0
Ejemplos =========Supongamos que
indices = [0, 2, -1, 1]
depth = 3
on_value = 5.0
off_value = 0.0
axis = -1
entonces la salida es `[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)
suponer que indices = [0, 2, -1, 1]
depth = 3
on_value = 0.0
off_value = 3.0
axis = 0
entonces la salida es `[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)
Supongamos que indices = [[0, 2], [1, -1]]
depth = 3
on_value = 1.0
off_value = 0.0
axis = -1
Luego de salida es `[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)
]
Clases anidadas
clase | OneHot.Options | Los atributos opcionales para OneHot |
Métodos públicos
Salida <U> | asOutput () Devuelve el identificador simbólico de un tensor. |
estáticas OneHot.Options | eje (eje largo) |
estática <U, T se extiende Número> OneHot <U> | |
Salida <U> | salida () El tensor one-hot. |
Métodos heredados
Métodos públicos
pública de salida <U> asOutput ()
Devuelve el identificador simbólico de un tensor.
Las entradas a las operaciones de TensorFlow son salidas de otra operación de TensorFlow. Este método se utiliza para obtener un identificador simbólico que representa el cálculo de la entrada.
public static OneHot.Options eje (eje largo)
Parámetros
eje | El eje a llenar (predeterminado: -1, un nuevo eje más interno). |
---|
public static OneHot <U> crear ( Alcance alcance, operando <T> índices, operando <Integer> profundidad, operando <U> onValue, operando <U> OffValue, Opciones ... Opciones)
Método de fábrica para crear una clase que envuelva una nueva operación OneHot.
Parámetros
alcance | alcance actual |
---|---|
índices | Un tensor de índices. |
profundidad | Un escalar que define la profundidad de una dimensión caliente. |
onValue | Un escalar que define el valor para completar la salida cuando `índices [j] = i`. |
offValue | Un escalar que define el valor para completar la salida cuando `índices [j]! = I`. |
opciones | lleva valores de atributos opcionales |
Devoluciones
- una nueva instancia de OneHot