تدفق التوتر:: العمليات:: DiagPart

#include <array_ops.h>

إرجاع الجزء القطري من الموتر.

ملخص

تقوم هذه العملية بإرجاع موتر مع الجزء diagonal من input . يتم حساب الجزء diagonal على النحو التالي:

افترض أن input له أبعاد [D1,..., Dk, D1,..., Dk] ، فإن الإخراج هو موتر من الرتبة k بأبعاد [D1,..., Dk] حيث:

diagonal[i1,..., ik] = input[i1, ..., ik, i1,..., ik] .

على سبيل المثال:

# '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]

الحجج:

  • النطاق: كائن النطاق
  • الإدخال: موتر الرتبة k حيث يكون k زوجيًا وليس صفرًا.

عائدات:

  • Output : القطر المستخرج.

البنائين والمدمرين

DiagPart (const :: tensorflow::Scope & scope, :: tensorflow::Input input)

الصفات العامة

diagonal
operation

الوظائف العامة

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

الصفات العامة

قطري

::tensorflow::Output diagonal

عملية

Operation operation

الوظائف العامة

DiagPart

 DiagPart(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input
)

العقدة

::tensorflow::Node * node() const

المشغل::tensorflow::الإدخال

 operator::tensorflow::Input() const

المشغل::tensorflow::الإخراج

 operator::tensorflow::Output() const