تينسورفلو :: العمليات :: SparseMatMul

#include <math_ops.h>

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

ملخص

يجب أن تكون المدخلات عبارة عن مصفوفات ثنائية الأبعاد ويجب أن يتطابق البعد الداخلي لـ "a" مع البعد الخارجي لـ "b". يجب أن يكون كل من "a" و "b" من Tensor وليس SparseTensor s. تم تحسين هذا المرجع للحالة التي يكون فيها واحد على الأقل من "أ" أو "ب" متفرقًا ، بمعنى أن لديهم نسبة كبيرة من القيم الصفرية. كان التعادل لاستخدام هذا مقابل ضرب مصفوفة كثيفة على منصة واحدة قيم صفرية بنسبة 30٪ في المصفوفة المتفرقة.

سيستفيد حساب التدرج لهذه العملية فقط من التباين في تدرج الإدخال عندما يأتي هذا التدرج من Relu .

الحجج:

عائدات:

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

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

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

السمات العامة

operation
product

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

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

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

AIsSparse (bool x)
BIsSparse (bool x)
TransposeA (bool x)
TransposeB (bool x)

الهياكل

tensorflow :: العمليات :: SparseMatMul :: Attrs

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

السمات العامة

عملية

Operation operation

منتج

::tensorflow::Output product

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

SparseMatMul

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

SparseMatMul

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

العقدة

::tensorflow::Node * node() const 

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

 operator::tensorflow::Input() const 

المشغل :: Tensorflow :: Output

 operator::tensorflow::Output() const 

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

الذكاء الاصطناعي

Attrs AIsSparse(
  bool x
)

تناثر

Attrs BIsSparse(
  bool x
)

تبديل

Attrs TransposeA(
  bool x
)

تبديل

Attrs TransposeB(
  bool x
)