aliran tensor:: operasi:: MatriksDiag
#include <array_ops.h>Mengembalikan tensor diagonal kumpulan dengan nilai diagonal kumpulan tertentu.
Ringkasan
Jika diberi diagonal , operasi ini mengembalikan tensor dengan diagonal dan yang lainnya diisi dengan nol. Diagonal dihitung sebagai berikut:
Asumsikan diagonal memiliki k dimensi [I, J, K, ..., N] , maka outputnya adalah tensor rank k+1 dengan dimensi [I, J, K, ..., N, N]` dimana:
output[i, j, k, ..., m, n] = 1{m=n} * diagonal[i, j, k, ..., n] .
Misalnya:
# '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)
Argumen:
- ruang lingkup: Objek Lingkup
- diagonal: Peringkat
k, di manak >= 1.
Pengembalian:
-
Output: Peringkatk+1, denganoutput.shape = diagonal.shape + [diagonal.shape[-1]].
Konstruktor dan Destruktor | |
|---|---|
MatrixDiag (const :: tensorflow::Scope & scope, :: tensorflow::Input diagonal) |
Atribut publik | |
|---|---|
operation | |
output | |
Fungsi publik | |
|---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const | |
Atribut publik
operasi
Operation operation
keluaran
::tensorflow::Output output
Fungsi publik
MatriksDiag
MatrixDiag( const ::tensorflow::Scope & scope, ::tensorflow::Input diagonal )
simpul
::tensorflow::Node * node() const
operator::tensorflow::Masukan
operator::tensorflow::Input() const
operator::tensorflow::Keluaran
operator::tensorflow::Output() const