Retorna um tensor one-hot.
Os locais representados por índices em `índices` assumem o valor` on_value`, enquanto todos os outros locais assumem o valor `off_value`.
Se os `índices` de entrada são classificação` N`, a saída terá classificação `N + 1`. O novo eixo é criado na dimensão` eixo` (padrão: o novo eixo é anexado no final).
Se `índices` for um escalar, a forma de saída será um vetor de comprimento` profundidade`.
Se `indices` é um vector de comprimento` features`, a forma de saída será:
features x depth if axis == -1
depth x features if axis == 0
Se `indices` é uma matriz (em lotes) com forma` [lote, caracteriza] `, a forma de saída 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
Exemplos =========Suponha-se que
indices = [0, 2, -1, 1]
depth = 3
on_value = 5.0
off_value = 0.0
axis = -1
então a saída é `[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)
supor que indices = [0, 2, -1, 1]
depth = 3
on_value = 0.0
off_value = 3.0
axis = 0
então a saída é `[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)
Suponhamos que indices = [[0, 2], [1, -1]]
depth = 3
on_value = 1.0
off_value = 0.0
axis = -1
Em seguida, a saída é `[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 aninhadas
classe | OneHot.Options | Atributos opcionais para OneHot |
Métodos Públicos
Output <U> | asOutput () Retorna o identificador simbólico de um tensor. |
estáticos OneHot.Options | eixo (eixo Long) |
estática <U, T estende Number> onehot <U> | |
Output <U> | saída () O tensor one-hot. |
Métodos herdados
Métodos Públicos
pública Output <U> asOutput ()
Retorna o identificador simbólico de um tensor.
As entradas para as operações do TensorFlow são saídas de outra operação do TensorFlow. Este método é usado para obter um identificador simbólico que representa o cálculo da entrada.
public static OneHot.Options eixos (eixo Long)
Parâmetros
eixo | O eixo a ser preenchido (padrão: -1, um novo eixo mais interno). |
---|
public static onehot <U> create ( Scope escopo, Operando <T> índices, Operando <Integer> profundidade, Operando <U> onValue, Operando <U> offValue, Options ... Opções)
Método de fábrica para criar uma classe envolvendo uma nova operação OneHot.
Parâmetros
alcance | escopo atual |
---|---|
índices | Um tensor de índices. |
profundidade | Um escalar que define a profundidade de uma dimensão quente. |
onValue | Um escalar que define o valor a ser preenchido na saída quando `índices [j] = i`. |
offValue | Um escalar que define o valor a ser preenchido na saída quando `índices [j]! = I`. |
opções | carrega valores de atributos opcionais |
Devoluções
- uma nova instância de OneHot