tensoreflusso:: ops:: DiagPart
#include <array_ops.h>Restituisce la parte diagonale del tensore.
Riepilogo
Questa operazione restituisce un tensore con la parte diagonal input . La parte diagonal si calcola come segue:
Supponiamo che input abbia dimensioni [D1,..., Dk, D1,..., Dk] , quindi l'output è un tensore di rango k con dimensioni [D1,..., Dk] dove:
diagonal[i1,..., ik] = input[i1, ..., ik, i1,..., ik] .
Per esempio:
# '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]
Argomenti:
- scope: un oggetto Scope
- input: tensore di rango k dove k è pari e non zero.
Resi:
-
Output: la diagonale estratta.
Costruttori e distruttori | |
|---|---|
DiagPart (const :: tensorflow::Scope & scope, :: tensorflow::Input input) |
Attributi pubblici | |
|---|---|
diagonal | |
operation | |
Funzioni pubbliche | |
|---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const | |
Attributi pubblici
diagonale
::tensorflow::Output diagonal
operazione
Operation operation
Funzioni pubbliche
DiagPart
DiagPart( const ::tensorflow::Scope & scope, ::tensorflow::Input input )
nodo
::tensorflow::Node * node() const
operatore::tensorflow::Input
operator::tensorflow::Input() const
operatore::tensorflow::Output
operator::tensorflow::Output() const