Gather

ชั้นเรียนสุดท้ายสาธารณะ รวบรวม

ล้อมตัวดำเนินการ 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

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