Retorna um tensor one-hot.
Os locais representados pelos índices em `indices` assumem o valor `on_value`, enquanto todos os outros locais assumem o valor `off_value`.
Se os `índices` de entrada forem de classificação `N`, a saída terá a classificação `N+1`. O novo eixo é criado na dimensão `axis` (padrão: o novo eixo é anexado no final).
Se `indices` 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
batch x features x depth if axis == -1
batch x depth x features if axis == 1
depth x batch x features if axis == 0
Suponha 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 aninhadas
aula | OneHot.Opções | Atributos opcionais para OneHot |
Constantes
Corda | OP_NAME | O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow |
Métodos Públicos
Saída <U> | asOutput () Retorna o identificador simbólico do tensor. |
OneHot.Options estático | eixo (eixo longo) |
estático <U estende TType > OneHot <U> | |
Saída <U> | saída () O tensor one-hot. |
Métodos herdados
Constantes
String final estática pública OP_NAME
O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow
Métodos Públicos
Saída pública <U> asOutput ()
Retorna o identificador simbólico do 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, Operando <? estende TNumber > índices, Operando < TInt32 > profundidade, Operando <U> onValue, Operando <U> offValue, Opções... opções)
Método de fábrica para criar uma classe que envolve uma nova operação OneHot.
Parâmetros
escopo | escopo atual |
---|---|
índices | Um tensor de índices. |
profundidade | Um escalar que define a profundidade de uma dimensão quente. |
emValor | Um escalar que define o valor a ser preenchido na 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