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:
Supongamos que la `entrada` 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
Cadena | OP_NOMBRE | El nombre de esta operación, como lo conoce el motor central de TensorFlow. |
Métodos públicos
Salida <T> | como salida () Devuelve el identificador simbólico del tensor. |
estático <T extiende TType > TensorDiagPart <T> | |
Salida <T> | diagonal () La diagonal extraída. |
Métodos heredados
Constantes
Cadena final estática pública OP_NAME
El nombre de esta operación, como la conoce el motor central de TensorFlow.
Métodos públicos
Salida pública <T> como Salida ()
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.
creación pública estática de TensorDiagPart <T> (alcance del alcance , entrada del operando <T>)
Método de fábrica para crear una clase que envuelve una nueva operación de TensorDiagPart.
Parámetros
alcance | alcance actual |
---|---|
aporte | Tensor de rango k donde k es par y no cero. |
Devoluciones
- una nueva instancia de TensorDiagPart