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

#include <math_ops.h>

اضرب المصفوفة "أ" في المصفوفة "ب".

ملخص

يجب أن تكون المدخلات مصفوفات ثنائية الأبعاد ويجب أن يتطابق البعد الداخلي لـ "a" (بعد تبديل موضعه إذا كان transpose_a صحيحًا) مع البعد الخارجي لـ "b" (بعد تبديل موضعه إذا كان transpose_b صحيحًا).

ملاحظة : يستخدم تطبيق kernel الافتراضي لـ MatMul على وحدات معالجة الرسومات cublas.

الحجج:

السمات الاختيارية (انظر Attrs ):

  • transpose_a: إذا كان صحيحًا، فسيتم نقل "a" قبل الضرب.
  • transpose_b: إذا كان صحيحًا، فسيتم نقل "b" قبل الضرب.

العوائد:

  • Output : موتر المنتج.

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

MatMul (const :: tensorflow::Scope & scope, :: tensorflow::Input a, :: tensorflow::Input b)
MatMul (const :: tensorflow::Scope & scope, :: tensorflow::Input a, :: tensorflow::Input b, const MatMul::Attrs & attrs)

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

operation
product

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

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

وظائف ثابتة العامة

TransposeA (bool x)
TransposeB (bool x)

الهياكل

Tensorflow:: ops:: MatMul:: Attrs

محددات السمات الاختيارية لـ MatMul .

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

عملية

Operation operation

منتج

::tensorflow::Output product

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

MatMul

 MatMul(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input a,
  ::tensorflow::Input b
)

MatMul

 MatMul(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input a,
  ::tensorflow::Input b,
  const MatMul::Attrs & attrs
)

العقدة

::tensorflow::Node * node() const 

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

 operator::tensorflow::Input() const 

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

 operator::tensorflow::Output() const 

وظائف ثابتة العامة

تبديل أ

Attrs TransposeA(
  bool x
)

تبديل ب

Attrs TransposeB(
  bool x
)