tensorflow :: ops :: MatrixDiag
#include <array_ops.h>
Mengembalikan tensor diagonal kelompok dengan nilai diagonal kelompok tertentu.
Ringkasan
Diberikan diagonal
, operasi ini mengembalikan tensor dengan diagonal
dan segala sesuatu yang diisi dengan nol. Diagonal dihitung sebagai berikut:
Asumsikan diagonal
memiliki dimensi k
[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]
.
Sebagai contoh:
# '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:
- scope: Objek Scope
- diagonal: Peringkat
k
, di manak >= 1
.
Pengembalian:
-
Output
: Peringkatk+1
, denganoutput.shape = diagonal.shape + [diagonal.shape[-1]]
.
Pembuat dan Penghancur | |
---|---|
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
MatrixDiag
MatrixDiag( const ::tensorflow::Scope & scope, ::tensorflow::Input diagonal )
simpul
::tensorflow::Node * node() const
operator :: tensorflow :: Input
operator::tensorflow::Input() const
operator :: tensorflow :: Keluaran
operator::tensorflow::Output() const