تدفق التوتر:: العمليات:: MatrixDiag
#include <array_ops.h>
تُرجع موترًا قطريًا مجمعًا بقيم قطرية مجمعة معينة.
ملخص
بالنظر إلى diagonal
، تُرجع هذه العملية موترًا diagonal
وكل شيء آخر مبطن بالأصفار. يتم حساب القطر على النحو التالي:
افترض أن diagonal
له أبعاد k
[I, J, K, ..., N]
، ثم يكون الناتج موترًا من الرتبة k+1
بأبعاد [I, J, K, ..., N, N]` حيث:
output[i, j, k, ..., m, n] = 1{m=n} * diagonal[i, j, k, ..., n]
.
على سبيل المثال:
# '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)
الحجج:
- النطاق: كائن النطاق
- قطري: الرتبة
k
, حيثk >= 1
.
العوائد:
-
Output
: الرتبةk+1
، معoutput.shape = diagonal.shape + [diagonal.shape[-1]]
.
البنائين والمدمرين | |
---|---|
MatrixDiag (const :: tensorflow::Scope & scope, :: tensorflow::Input diagonal) |
الوظائف العامة | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
الصفات العامة
عملية
Operation operation
الإخراج
::tensorflow::Output output
الوظائف العامة
MatrixDiag
MatrixDiag( const ::tensorflow::Scope & scope, ::tensorflow::Input diagonal )
العقدة
::tensorflow::Node * node() const
المشغل::tensorflow::الإدخال
operator::tensorflow::Input() const
المشغل::tensorflow::الإخراج
operator::tensorflow::Output() const