יום הקהילה ML הוא 9 בנובמבר! הצטרפו אלינו עדכונים מ- TensorFlow, JAX, ועוד למידע נוסף

זרימת טנסור :: אופ :: MatMul

#include <math_ops.h>

הכפל את המטריצה ​​"a" במטריצה ​​"b".

סיכום

הקלטים חייבים להיות מטריצות דו ממדיות והמימד הפנימי של "a" (לאחר שהועבר אם transpos_a נכון) חייב להתאים למימד החיצוני של "b" (לאחר שהועבר אם transpone_b נכון).

הערה : יישום הקרנל המוגדר כברירת מחדל עבור MatMul ב- GPUs משתמש בקובלות.

טענות:

  • היקף: אובייקט Scope

תכונות אופציונליות (ראה 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 

אופרטור :: זרימת טנסור :: קלט

 operator::tensorflow::Input() const 

אופרטור :: זרימת טנסור :: פלט

 operator::tensorflow::Output() const 

פונקציות סטטיות ציבוריות

TransposeA

Attrs TransposeA(
  bool x
)

TransposeB

Attrs TransposeB(
  bool x
)