tensorflow::ops::BatchMatMul

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

สรุป

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

เทนเซอร์อินพุต x และ y เป็นเทนเซอร์ 2 มิติขึ้นไปที่มีรูปร่าง [..., r_x, c_x] และ [..., r_y, c_y]

เทนเซอร์เอาต์พุตเป็นเทนเซอร์ 2 มิติขึ้นไปที่มีรูปร่าง [..., 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[..., :, :])

ข้อโต้แย้ง:

• ขอบเขต: วัตถุ ขอบเขต
• x: 2 มิติขึ้นไปที่มีรูปร่าง [..., r_x, c_x]
• y: 2 มิติขึ้นไปที่มีรูปร่าง [..., r_y, c_y]

คุณลักษณะเสริม (ดูที่ Attrs ):

• adj_x: ถ้า True จะทำการผกผันส่วนย่อยของ x ค่าเริ่มต้นคือ False
• adj_y: ถ้า True จะทำการผกผันส่วนย่อยของ y ค่าเริ่มต้นคือ False

การส่งคืนสินค้า:

• Output : 3 มิติขึ้นไปที่มีรูปร่าง [..., r_o, c_o]

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

Operation operation

::tensorflow::Output output

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

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

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const 

Attrs AdjX(
  bool x
)

Attrs AdjY(
  bool x
)