Devuelve la parte diagonal del tensor.
Esta operación devuelve un tensor con la parte "diagonal" de la "entrada". La parte `diagonal` se calcula de la siguiente manera:
Suponga que "input" tiene dimensiones "[D1, ..., Dk, D1, ..., Dk]", entonces la salida es un tensor de rango "k" con dimensiones "[D1, ..., Dk]" dónde:
`diagonal [i1, ..., ik] = entrada [i1, ..., ik, i1, ..., ik]`.
Por ejemplo:
# '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
Cuerda | OP_NAME | El nombre de esta operación, como lo conoce el motor central de TensorFlow |
Métodos públicos
Salida <T> | asOutput () Devuelve el identificador simbólico del tensor. |
estática <T se extiende Ttype > TensorDiagPart <T> | |
Salida <T> | diagonal () La diagonal extraída. |
Métodos heredados
Constantes
OP_NAME pública final static String
El nombre de esta operación, como lo conoce el motor central de TensorFlow
Métodos públicos
pública de salida <T> asOutput ()
Devuelve el identificador simbólico del tensor.
Las entradas a las operaciones de TensorFlow son salidas de otra operación de TensorFlow. Este método se utiliza para obtener un identificador simbólico que representa el cálculo de la entrada.
public static TensorDiagPart <T> crear ( Alcance alcance, operando <T> entrada)
Método de fábrica para crear una clase que envuelva una nueva operación TensorDiagPart.
Parámetros
alcance | alcance actual |
---|---|
aporte | Tensor de rango k donde k es par y no cero. |
Devoluciones
- una nueva instancia de TensorDiagPart