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