جریان تنسور:: عملیات:: MatMul
#include <math_ops.h>
ماتریس "a" را در ماتریس "b" ضرب کنید .
خلاصه
ورودی ها باید ماتریس های دو بعدی باشند و بعد داخلی "a" (پس از انتقال اگر transpose_a درست باشد) باید با بعد بیرونی "b" مطابقت داشته باشد (پس از انتقال اگر transposed_b درست باشد).
توجه : اجرای هسته پیشفرض برای MatMul روی پردازندههای گرافیکی از cublas استفاده میکند.
استدلال ها:
- scope: یک شی 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) |
توابع عمومی | |
---|---|
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 )