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

#include <math_ops.h>

ضرب شرائح اثنين من الموتدين على دفعات.

ملخص

مضاعفة جميع شرائح Tensor x و y (يمكن عرض كل شريحة كعنصر من عناصر دفعة)، وترتيب النتائج الفردية في موتر إخراج واحد بنفس حجم الدفعة. يمكن أن تكون كل شريحة من الشرائح الفردية متجاورة بشكل اختياري (مجاورة المصفوفة تعني تبديلها وتصريفها) قبل الضرب عن طريق تعيين علامة adj_x أو adj_y على True ، والتي تكون افتراضيًا False .

موترات الإدخال x و y هي ثنائية الأبعاد أو أعلى بالشكل [..., r_x, c_x] و [..., r_y, c_y] .

موتر الإخراج ثنائي الأبعاد أو أعلى بالشكل [..., r_o, c_o] حيث:

r_o = c_x if adj_x else r_x
c_o = r_y if adj_y else c_y

يتم حسابه على النحو التالي:

output[..., :, :] = matrix(x[..., :, :]) * matrix(y[..., :, :])

ملاحظة : يدعم BatchMatMulV2 البث في أبعاد الدُفعة. المزيد عن البث هنا .

الحجج:

  • النطاق: كائن النطاق
  • x: ثنائي الأبعاد أو أعلى بالشكل [..., r_x, c_x] .
  • y: ثنائي الأبعاد أو أعلى بالشكل [..., r_y, c_y] .

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

  • adj_x: إذا كان True ، قم بضم شرائح x إلى جوارها. الإعدادات الافتراضية إلى False .
  • adj_y: إذا كان True ، بجوار شرائح y . الإعدادات الافتراضية إلى False .

عائدات:

  • Output : ثلاثي الأبعاد أو أعلى بالشكل [..., r_o, c_o]

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

BatchMatMulV2 (const :: tensorflow::Scope & scope, :: tensorflow::Input x, :: tensorflow::Input y)
BatchMatMulV2 (const :: tensorflow::Scope & scope, :: tensorflow::Input x, :: tensorflow::Input y, const BatchMatMulV2::Attrs & attrs)

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

operation
output

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

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

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

AdjX (bool x)
AdjY (bool x)

الهياكل

Tensorflow:: ops:: BatchMatMulV2:: Attrs

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

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

عملية

Operation operation

انتاج

::tensorflow::Output output

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

BatchMatMulV2

 BatchMatMulV2(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input x,
  ::tensorflow::Input y
)

BatchMatMulV2

 BatchMatMulV2(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input x,
  ::tensorflow::Input y,
  const BatchMatMulV2::Attrs & attrs
)

العقدة

::tensorflow::Node * node() const 

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

 operator::tensorflow::Input() const 

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

 operator::tensorflow::Output() const 

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

AdjX

Attrs AdjX(
  bool x
)

أدجي

Attrs AdjY(
  bool x
)