เทนเซอร์โฟลว์ :: ops :: MatMul

#include <math_ops.h>

คูณ เมทริกซ์ "a" ด้วยเมทริกซ์ "b"

สรุป

อินพุตต้องเป็นเมทริกซ์สองมิติและมิติด้านในของ "a" (หลังจากที่ถูกเปลี่ยนหาก transpose_a เป็นจริง) ต้องตรงกับมิติภายนอกของ "b" (หลังจากถูกเปลี่ยนหาก transposed_b เป็นจริง)

หมายเหตุ : การใช้งานเคอร์เนลเริ่มต้นสำหรับ MatMul บน GPU จะใช้ cublas

อาร์กิวเมนต์:

แอตทริบิวต์เสริม (ดู 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)

คุณลักษณะสาธารณะ

operation
product

หน้าที่สาธารณะ

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
)

โหนด

::tensorflow::Node * node() const 

ตัวดำเนินการ :: tensorflow :: อินพุต

 operator::tensorflow::Input() const 

ตัวดำเนินการ :: tensorflow :: เอาต์พุต

 operator::tensorflow::Output() const 

ฟังก์ชันคงที่สาธารณะ

ทรานสโพส

Attrs TransposeA(
  bool x
)

ทรานสโพส

Attrs TransposeB(
  bool x
)