Retorna um tensor diagonal com determinados valores diagonais.
Dada uma `diagonal`, esta operação retorna um tensor com a `diagonal` e todo o resto preenchido com zeros. A diagonal é calculada da seguinte forma:
Suponha que `diagonal` tenha dimensões [D1,..., Dk], então a saída é um tensor de classificação 2k com dimensões [D1,..., Dk, D1,..., Dk] onde:
`output[i1,..., ik, i1,..., ik] = diagonal[i1, ..., ik]` e 0 em qualquer outro lugar.
Por exemplo:
# 'diagonal' is [1, 2, 3, 4]
tf.diag(diagonal) ==> [[1, 0, 0, 0]
[0, 2, 0, 0]
[0, 0, 3, 0]
[0, 0, 0, 4]]
Constantes
Corda | OP_NAME | O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow |
Métodos Públicos
Saída <T> | asOutput () Retorna o identificador simbólico do tensor. |
estático <T estende TType > TensorDiag <T> | |
Saída <T> | saída () |
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 <T> 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.
public static TensorDiag <T> create (escopo do escopo , operando <T> diagonal)
Método de fábrica para criar uma classe que envolve uma nova operação TensorDiag.
Parâmetros
escopo | escopo atual |
---|---|
diagonal | Tensor de classificação k onde k é no máximo 1. |
Devoluções
- uma nova instância do TensorDiag