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

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

#include <math_ops.h>

מכפיל פרוסות של שני טנזורים במנות.

סיכום

מכפיל את כל פרוסות ה- Tensor x ו- y (ניתן לראות כל פרוסה כאלמנט של אצווה), ומסדר את התוצאות הבודדות בטנסור פלט יחיד באותו גודל אצווה. ניתן לחבר לכל אחת מהפרוסות האינדיבידואליות (לצרף מטריצה ​​פירושה להעביר אותה adj_x ) לפני הכפל על ידי הגדרת הדגל adj_y או 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 תומך בשידור במידות האצווה. עוד על שידור כאן .

טענות:

  • היקף: אובייקט Scope
  • x: 2-D ומעלה עם צורה [..., r_x, c_x] .
  • y: 2-D ומעלה עם צורה [..., 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 

אופרטור :: זורם טנסור :: קלט

 operator::tensorflow::Input() const 

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

 operator::tensorflow::Output() const 

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

AdjX

Attrs AdjX(
  bool x
)

AdjY

Attrs AdjY(
  bool x
)