Retorna a parte diagonal do tensor.
Esta operação retorna um tensor com a parte `diagonal` da `entrada`. A parte `diagonal` é calculada da seguinte forma:
Suponha que `entrada` tenha dimensões `[D1,..., Dk, D1,..., Dk]`, então a saída é um tensor de classificação `k` com dimensões `[D1,..., Dk]` onde:
`diagonal[i1,..., ik] = entrada[i1, ..., ik, i1,..., ik]`.
Por exemplo:
# 'input' is [[1, 0, 0, 0]
[0, 2, 0, 0]
[0, 0, 3, 0]
[0, 0, 0, 4]]
tf.diag_part(input) ==> [1, 2, 3, 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 > TensorDiagPart <T> | |
| Saída <T> | diagonal () A diagonal extraí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 TensorDiagPart <T> create (escopo do escopo , entrada Operand <T>)
Método de fábrica para criar uma classe que envolve uma nova operação TensorDiagPart.
Parâmetros
| escopo | escopo atual |
|---|---|
| entrada | Tensor de classificação k onde k é par e não zero. |
Devoluções
- uma nova instância do TensorDiagPart