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

#include <math_ops.h>

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

ملخص

يجب أن تكون المدخلات مصفوفات ثنائية الأبعاد ويجب أن يتطابق البعد الداخلي لـ "أ" مع البعد الخارجي لـ "ب". يجب أن يكون كل من "a" و"b" Tensor وليس SparseTensor s. تم تحسين هذه العملية للحالة التي يكون فيها واحد على الأقل من "a" أو "b" متناثرًا، بمعنى أن لديهم نسبة كبيرة من القيم الصفرية. وكان التعادل لاستخدام هذا مقابل مصفوفة كثيفة تتضاعف على منصة واحدة هو 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:: ops:: SparseMatMul:: Attrs

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

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

عملية

Operation operation

منتج

::tensorflow::Output product

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

SparsMatMul

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

SparsMatMul

 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::الإخراج

 operator::tensorflow::Output() const 

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

AIsSparse

Attrs AIsSparse(
  bool x
)

BISSparse

Attrs BIsSparse(
  bool x
)

تبديل أ

Attrs TransposeA(
  bool x
)

تبديل ب

Attrs TransposeB(
  bool x
)