tensorflow :: ops :: DiagPart
#include <array_ops.h>
Mengembalikan bagian diagonal tensor.
Ringkasan
Operasi ini mengembalikan tensor dengan input
bagian diagonal
. Bagian diagonal
dihitung sebagai berikut:
Asumsikan input
memiliki dimensi [D1,..., Dk, D1,..., Dk]
, maka outputnya berupa tensor rank k
dengan dimensi [D1,..., Dk]
dimana:
diagonal[i1,..., ik] = input[i1, ..., ik, i1,..., ik]
.
Sebagai contoh:
# '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]
Argumen:
- scope: Objek Scope
- masukan: Rangking k tensor di mana k genap dan bukan nol.
Pengembalian:
-
Output
: Diagonal yang diekstraksi.
Pembuat dan Penghancur | |
---|---|
DiagPart (const :: tensorflow::Scope & scope, :: tensorflow::Input input) |
Atribut publik | |
---|---|
diagonal | |
operation |
Fungsi publik | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
Atribut publik
diagonal
::tensorflow::Output diagonal
operasi
Operation operation
Fungsi publik
DiagPart
DiagPart( const ::tensorflow::Scope & scope, ::tensorflow::Input input )
simpul
::tensorflow::Node * node() const
operator :: tensorflow :: Input
operator::tensorflow::Input() const
operator :: tensorflow :: Keluaran
operator::tensorflow::Output() const