Retorna um tensor one-hot.
Os locais representados por índices em `indices` assumem o valor `on_value`, enquanto todos os outros locais assumem o valor `off_value`.
Se a entrada `indices` for rank `N`, a saída terá rank `N+1`, O novo eixo é criado na dimensão `axis` (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` for um vetor de comprimento `features`, a forma de saída será:
features x depth if axis == -1
depth x features if axis == 0
Se `indices` for uma matriz (lote) com a forma `[lote, features]`, 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 que
indices = [0, 2, -1, 1]
depth = 3
on_value = 5.0
off_value = 0.0
axis = -1
Então a saída seja `[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)
Suponha que indices = [0, 2, -1, 1]
depth = 3
on_value = 0.0
off_value = 3.0
axis = 0
Então a saída seja `[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)
Suponha que indices = [[0, 2], [1, -1]]
depth = 3
on_value = 1.0
off_value = 0.0
axis = -1
Então a saída seja `[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
aula | OneHot.Options | Atributos opcionais para OneHot |
Métodos públicos
Saída <U> | comoSaída () Retorna o identificador simbólico de um tensor. |
estático OneHot.Options | eixo (eixo longo) |
estático <U, T estende Número> OneHot <U> | |
Saída <U> | saída () O tensor one-hot. |
Métodos Herdados
Métodos públicos
Public Output <U> asOutput ()
Retorna o identificador simbólico de um tensor.
As entradas para 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.
Eixo público estático OneHot.Options ( eixo longo)
Parâmetros
eixo | O eixo a ser preenchido (padrão: -1, um novo eixo mais interno). |
---|
public static OneHot <U> create ( Escopo do escopo, Índices do Operando <T>, Profundidade do Operando <Integer>, Operando <U> onValue, Operando <U> offValue, Options... options)
Método de fábrica para criar uma classe que envolve 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 para preencher a saída quando `indices[j] = i`. |
offValor | Um escalar que define o valor a ser preenchido na saída quando `indices[j] != i`. |
opções | carrega valores de atributos opcionais |
Devoluções
- uma nova instância do OneHot