ชั้นเรียนสุดท้ายสาธารณะ รวบรวม
ล้อมตัวดำเนินการ XLA Gather ที่บันทึกไว้ที่
https://www.tensorflow.org/xla/operation_semantics#gather
ค่าคงที่
สตริง | OP_NAME | ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow |
วิธีการสาธารณะ
เอาท์พุต <T> | เป็นเอาท์พุต () ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์ |
คงที่ <T ขยาย TType , U ขยาย TNumber > รวบรวม <T> | สร้าง (ขอบเขต ขอบเขต ตัวดำเนิน การ <T> ตัวถูก ดำเนินการ <U> startIndices ตัวดำเนินการ <U> ชิ้นขนาด ขนาดสตริงตัวเลข ดัชนีบูลีนAreSorted) วิธีการจากโรงงานเพื่อสร้างคลาสที่รวบรวมการดำเนินการรวบรวมใหม่ |
เอาท์พุต <T> | เอาท์พุท () |
วิธีการสืบทอด
ค่าคงที่
สตริงสุดท้ายแบบคงที่สาธารณะ OP_NAME
ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow
ค่าคงที่: "XlaGather"
วิธีการสาธารณะ
เอาท์ พุท สาธารณะ <T> asOutput ()
ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์
อินพุตสำหรับการดำเนินการ TensorFlow คือเอาต์พุตของการดำเนินการ TensorFlow อื่น วิธีการนี้ใช้เพื่อรับหมายเลขอ้างอิงสัญลักษณ์ที่แสดงถึงการคำนวณอินพุต
สาธารณะคง รวบรวม <T> สร้าง (ขอบเขต ขอบเขต ตัวดำเนิน การ <T> ตัวถูกดำเนินการ ตัวดำเนินการ <U> startIndices ตัวดำเนินการ <U> ชิ้นขนาด ขนาดสตริงหมายเลข ดัชนีบูลีนจะเรียงลำดับ)
วิธีการจากโรงงานเพื่อสร้างคลาสที่รวบรวมการดำเนินการรวบรวมใหม่
พารามิเตอร์
ขอบเขต | ขอบเขตปัจจุบัน |
---|---|
ตัวถูกดำเนินการ | อาร์เรย์ที่เรารวบรวมมา |
startIndices | อาร์เรย์ที่มีดัชนีเริ่มต้นของชิ้นที่เรารวบรวม |
ชิ้นขนาด | Slice_sizes[i] คือขอบเขตของชิ้นในมิติ i |
มิติตัวเลข | โปรโต xla::GatherDimensionNumbers ที่ต่อเนื่องกัน |
ดัชนีจะถูกจัดเรียง | บูลีนที่ระบุว่ามีการจัดเรียงดัชนีหรือไม่ |
การส่งคืน
- ตัวอย่างใหม่ของ Gather