เทนเซอร์โฟลว์:: ปฏิบัติการ:: BatchMatMulV2

คูณเทนเซอร์สองตัวเป็นชุด

สรุป

คูณชิ้นส่วนทั้งหมดของ Tensor x และ y (แต่ละชิ้นสามารถดูได้เป็นองค์ประกอบของชุด) และจัดเรียงผลลัพธ์แต่ละรายการในเทนเซอร์เอาต์พุตเดี่ยวที่มีขนาดชุดเดียวกัน แต่ละชิ้นสามารถเลือกที่จะติดกัน (เพื่อติดเมทริกซ์หมายถึงการย้ายและคอนจูเกต) ก่อนการคูณโดยการตั้งค่าแฟล็ก adj_x หรือ adj_y เป็น True ซึ่งตามค่าเริ่มต้นจะเป็น False

เทนเซอร์อินพุต x และ y เป็น 2-D หรือสูงกว่าโดยมีรูปร่าง [..., r_x, c_x] และ [..., r_y, c_y]

เอาท์พุตเทนเซอร์เป็น 2-D หรือสูงกว่าที่มีรูปร่าง [..., r_o, c_o] โดยที่:

r_o = c_x if adj_x else r_x
c_o = r_y if adj_y else c_y

คำนวณเป็น:

output[..., :, :] = matrix(x[..., :, :]) * matrix(y[..., :, :])

หมายเหตุ : BatchMatMulV2 รองรับการออกอากาศในมิติแบทช์ เพิ่มเติมเกี่ยวกับการออกอากาศ ที่นี่

ข้อโต้แย้ง:

• ขอบเขต: วัตถุ ขอบเขต
• x: 2-D หรือสูงกว่าที่มีรูปร่าง [..., r_x, c_x]
• y: 2-D หรือสูงกว่าที่มีรูปร่าง [..., r_y, c_y]

แอ็ตทริบิวต์ทางเลือก (ดู Attrs ):

• adj_x: ถ้าเป็น True ให้ติดส่วนของ x ค่าเริ่มต้นเป็น False
• adj_y: หาก True ให้ติดส่วนของ y ค่าเริ่มต้นเป็น False

ผลตอบแทน:

• Output : 3-D หรือสูงกว่าที่มีรูปร่าง [..., r_o, c_o]

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

Operation operation

::tensorflow::Output output

 BatchMatMulV2(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input x,
  ::tensorflow::Input y
)

 BatchMatMulV2(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input x,
  ::tensorflow::Input y,
  const BatchMatMulV2::Attrs & attrs
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const 

Attrs AdjX(
  bool x
)

Attrs AdjY(
  bool x
)