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 |
Constantes
Cuerda | OP_NAME | El nombre de esta operación, como lo conoce el motor central de TensorFlow |
Métodos públicos
Salida <U> | asOutput () Devuelve el identificador simbólico del tensor. |
estáticas OneHot.Options | eje (eje largo) |
estática <U se extiende Ttype > OneHot <U> | |
Salida <U> | salida () El tensor one-hot. |
Métodos heredados
Constantes
OP_NAME pública final static String
El nombre de esta operación, como lo conoce el motor central de TensorFlow
Métodos públicos
pública de salida <U> asOutput ()
Devuelve el identificador simbólico del 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 <? extiende TNumber > índices, operando < TInt32 > 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