BroadcastHelper

BroadcastHelper คลาสสุดท้ายสาธารณะ

ตัวช่วยดำเนินการออกอากาศแบบ XLA

ออกอากาศ `lhs` และ `rhs` ไปยังอันดับเดียวกัน โดยเพิ่มขนาด 1 มิติให้กับ `lhs` และ `rhs` ใดก็ตามที่มีอันดับต่ำกว่า โดยใช้กฎการออกอากาศของ XLA สำหรับตัวดำเนินการไบนารี

ค่าคงที่

สตริง OP_NAME ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow

วิธีการสาธารณะ

คงที่ <T ขยาย TType > BroadcastHelper <T>
สร้าง (ขอบเขต ขอบเขต ตัวดำเนินการ <T> lhs ตัวดำเนินการ <T> rhs ตัวดำเนินการ <? ขยาย TNumber > BroadcastDims)
วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ BroadcastHelper ใหม่
เอาท์พุต <T>
lhsเอาท์พุต ()
เทนเซอร์ LHS ที่ออกอากาศ
เอาท์พุต <T>
rhsเอาท์พุต ()
เทนเซอร์ RHS ที่ออกอากาศ

วิธีการสืบทอด

ค่าคงที่

สตริงสุดท้ายแบบคงที่สาธารณะ OP_NAME

ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow

ค่าคงที่: "XlaBroadcastHelper"

วิธีการสาธารณะ

BroadcastHelper แบบคงที่สาธารณะ <T> สร้าง (ขอบเขต ขอบเขต , ตัวดำเนินการ <T> lhs, ตัวดำเนินการ <T> rhs, ตัวดำเนินการ <? ขยาย TNumber > BroadcastDims)

วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ BroadcastHelper ใหม่

พารามิเตอร์
ขอบเขต ขอบเขตปัจจุบัน
นะ เทนเซอร์อินพุต LHS
Rhs เทนเซอร์อินพุต RHS
ออกอากาศ Dims ข้อกำหนดมิติการออกอากาศแบบ XLA
การส่งคืน
  • อินสแตนซ์ใหม่ของ BroadcastHelper

สาธารณะ เอาท์พุท <T> lhsOutput ()

เทนเซอร์ LHS ที่ออกอากาศ

เอาท์ พุท สาธารณะ <T> rhsOutput ()

เทนเซอร์ RHS ที่ออกอากาศ