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 |
Métodos Públicos
Saída <U> | asOutput () Retorna o identificador simbólico de um tensor. |
OneHot.Options estático | eixo (eixo longo) |
estático <U, T estende número> OneHot <U> | create (escopo do escopo , índices do operando <T>, profundidade do operando <inteiro>, 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. |
Saída <U> | saída () O tensor one-hot. |
Métodos herdados
Métodos Públicos
Saída pública <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 <inteiro>, 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