aliran tensor:: operasi:: Bagian Diag
#include <array_ops.h>
Mengembalikan bagian diagonal tensor.
Ringkasan
Operasi ini mengembalikan tensor dengan bagian diagonal
dari input
. Bagian diagonal
dihitung sebagai berikut:
Asumsikan input
berdimensi [D1,..., Dk, D1,..., Dk]
, maka outputnya berupa tensor rank k
dengan dimensi [D1,..., Dk]
dimana:
diagonal[i1,..., ik] = input[i1, ..., ik, i1,..., ik]
.
Misalnya:
# '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:
- ruang lingkup: Objek Lingkup
- masukan: Peringkat k tensor dengan k genap dan bukan nol.
Pengembalian:
-
Output
: Diagonal yang diekstraksi.
Konstruktor dan Destruktor | |
---|---|
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
Bagian Diag
DiagPart( const ::tensorflow::Scope & scope, ::tensorflow::Input input )
simpul
::tensorflow::Node * node() const
operator::tensorflow::Masukan
operator::tensorflow::Input() const
operator::tensorflow::Keluaran
operator::tensorflow::Output() const