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

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

#include <sparse_ops.h>

הכפל את SparseTensor (בדרגה 2) "A" לפי מטריצה ​​צפופה "B".

סיכום

לא מתבצעת בדיקת תוקף במדדים של A. עם זאת, תבנית הקלט הבאה מומלצת להתנהגות אופטימלית:

אם adjoint_a == false: יש למיין את A לפי סדר הגדלה לקסיקוגרפית. השתמש ב- SparseReorder אם אינך בטוח. אם adjoint_a == true: יש למיין את A לפי סדר הגדלת ממד 1 (כלומר, סדר "עמודות עיקריות" במקום סדר "שורה עיקרית").

טענות:

  • היקף: אובייקט Scope
  • a_indices: 2-D. indices של SparseTensor , גודל [nnz, 2] מטריקס.
  • a_values: 1-D. values של SparseTensor , גודל [nnz] .
  • a_shape: 1-D. shape ה- SparseTensor , גודל [2] וקטור.
  • b: 2-D. מטריקס צפוף.

תכונות אופציונליות (ראה Attrs ):

  • adjoint_a: השתמש בתוספת A בכפלת המטריצה. אם A מורכב, זה טרנספוזיציה (conj (A)). אחרת זה להעביר (A).
  • adjoint_b: השתמש בתוספת של B בכפלת המטריצה. אם B מורכב, זה טרנספוזיציה (conj (B)). אחרת זה להעביר (B).

החזרות:

  • Output : טנזור המוצר.

קונסטרוקטורים ומשחתנים

SparseTensorDenseMatMul (const :: tensorflow::Scope & scope, :: tensorflow::Input a_indices, :: tensorflow::Input a_values, :: tensorflow::Input a_shape, :: tensorflow::Input b)
SparseTensorDenseMatMul (const :: tensorflow::Scope & scope, :: tensorflow::Input a_indices, :: tensorflow::Input a_values, :: tensorflow::Input a_shape, :: tensorflow::Input b, const SparseTensorDenseMatMul::Attrs & attrs)

תכונות ציבוריות

operation
product

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

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

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

AdjointA (bool x)
AdjointB (bool x)

סטרוקטורים

tensorflow :: ops :: SparseTensorDenseMatMul :: Attrs

קובעי תכונות אופציונליים עבור SparseTensorDenseMatMul .

תכונות ציבוריות

מבצע

Operation operation

מוצר

::tensorflow::Output product

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

SparseTensorDenseMatMul

 SparseTensorDenseMatMul(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input a_indices,
  ::tensorflow::Input a_values,
  ::tensorflow::Input a_shape,
  ::tensorflow::Input b
)

SparseTensorDenseMatMul

 SparseTensorDenseMatMul(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input a_indices,
  ::tensorflow::Input a_values,
  ::tensorflow::Input a_shape,
  ::tensorflow::Input b,
  const SparseTensorDenseMatMul::Attrs & attrs
)

צוֹמֶת

::tensorflow::Node * node() const 

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

 operator::tensorflow::Input() const 

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

 operator::tensorflow::Output() const 

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

צמוד A.

Attrs AdjointA(
  bool x
)

צמוד B.

Attrs AdjointB(
  bool x
)