Devuelve un tensor diagonal con valores diagonales dados.
Dada una "diagonal", esta operación devuelve un tensor con la "diagonal" y todo lo demás relleno con ceros. La diagonal se calcula de la siguiente manera:
Supongamos que `diagonal` tiene dimensiones [D1,..., Dk], entonces la salida es un tensor de rango 2k con dimensiones [D1,..., Dk, D1,..., Dk] donde:
`salida[i1,..., ik, i1,..., ik] = diagonal[i1,..., ik]` y 0 en todos los demás.
Por ejemplo:
# '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
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 > TensorDiag <T> | |
Salida <T> | producción () |
Métodos heredados
Constantes
Cadena final estática pública OP_NAME
El nombre de esta operación, como lo 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.
TensorDiag <T> estático público crear (alcance alcance , operando <T> diagonal)
Método de fábrica para crear una clase que envuelve una nueva operación de TensorDiag.
Parámetros
alcance | alcance actual |
---|---|
diagonal | Tensor de rango k donde k es como máximo 1. |
Devoluciones
- una nueva instancia de TensorDiag