tensorflow :: ops :: MatrixDiag
#include <array_ops.h>
Restituisce un tensore diagonale in batch con un dato valore diagonale in batch.
Sommario
Data una diagonal
, questa operazione restituisce un tensore con la diagonal
e tutto il resto imbottito di zeri. La diagonale viene calcolata come segue:
Supponiamo che la diagonal
abbia k
dimensioni [I, J, K, ..., N]
, quindi l'uscita è un tensore di rango k+1
con dimensioni [I, J, K, ..., N, N] `dove:
output[i, j, k, ..., m, n] = 1{m=n} * diagonal[i, j, k, ..., n]
.
Per esempio:
# 'diagonal' is [[1, 2, 3, 4], [5, 6, 7, 8]]
and diagonal.shape = (2, 4)
tf.matrix_diag(diagonal) ==> [[[1, 0, 0, 0] [0, 2, 0, 0] [0, 0, 3, 0] [0, 0, 0, 4]], [[5, 0, 0, 0] [0, 6, 0, 0] [0, 0, 7, 0] [0, 0, 0, 8]]]
which has shape (2, 4, 4)
Argomenti:
- scope: un oggetto Scope
- diagonale: rango
k
, dovek >= 1
.
Ritorna:
-
Output
: Rankk+1
, conoutput.shape = diagonal.shape + [diagonal.shape[-1]]
.
Costruttori e distruttori | |
---|---|
MatrixDiag (const :: tensorflow::Scope & scope, :: tensorflow::Input diagonal) |
Attributi pubblici | |
---|---|
operation | |
output |
Funzioni pubbliche | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
Attributi pubblici
operazione
Operation operation
produzione
::tensorflow::Output output
Funzioni pubbliche
MatrixDiag
MatrixDiag( const ::tensorflow::Scope & scope, ::tensorflow::Input diagonal )
nodo
::tensorflow::Node * node() const
operator :: tensorflow :: Input
operator::tensorflow::Input() const
operator :: tensorflow :: Output
operator::tensorflow::Output() const