tensorflow::ops::BatchMatMul
#include <math_ops.h>คูณส่วนย่อยของเทนเซอร์สองตัวเป็นชุดๆ
สรุป
ฟังก์ชันนี้จะคูณส่วนย่อยทั้งหมดของ 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]
ตัวสร้างและตัวทำลาย | |
|---|---|
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) |
คุณลักษณะสาธารณะ | |
|---|---|
operation | |
output | |
หน้าที่สาธารณะ | |
|---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const | |
ฟังก์ชันคงที่สาธารณะ | |
|---|---|
AdjX (bool x) | |
AdjY (bool x) | |
โครงสร้าง | |
|---|---|
| tensorflow::ops::BatchMatMul::Attrs | ตัวเลือกการตั้งค่าแอตทริบิวต์สำหรับ BatchMatMul |
คุณลักษณะสาธารณะ
การดำเนินการ
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 )