นำจำนวนองค์ประกอบที่กำหนดจากบาเรีย
การดำเนินการนี้เชื่อมเทนเซอร์ขององค์ประกอบที่เสร็จสมบูรณ์ตามมิติที่ 0 เพื่อสร้างเมตริกซ์ส่วนประกอบเดียว
ส่วนประกอบจะออกมาจากบาเรียเมื่อสร้างเสร็จ และเรียงลำดับให้อยู่ในบาเรีย เอาต์พุตดัชนีให้ข้อมูลเกี่ยวกับแบทช์ที่แต่เดิมแต่ละองค์ประกอบถูกแทรกเข้าไปในอุปสรรค
คลาสที่ซ้อนกัน
ระดับ | BarrierTakeMany.Options | คุณลักษณะที่จำเป็นสำหรับการ BarrierTakeMany |
วิธีการสาธารณะ
คง BarrierTakeMany.Options | allowSmallBatch (บูลีน allowSmallBatch) |
คง BarrierTakeMany | สร้าง ( ขอบเขต ขอบเขต Operand <String> จับ ตัวถูกดำเนินการ <Integer> numElements รายการ <ชั้น <? >> componentTypes, ตัวเลือก ... ตัวเลือก) วิธีการจากโรงงานเพื่อสร้างคลาสที่ปิดการดำเนินการ BarrierTakeMany ใหม่ |
เอาท์พุท <ยาว> | ดัชนี () เทนเซอร์หนึ่งมิติของดัชนี มีความยาว num_elems |
เอาท์พุท <String> | คีย์ () เมตริกซ์หนึ่งมิติของคีย์ โดยมีความยาว num_elements |
คง BarrierTakeMany.Options | timeoutMs (Long timeoutMs) |
รายการ < เอาท์พุท <? >> | ค่า () หนึ่งเทนเซอร์ทุกมิติต่อส่วนประกอบในองค์ประกอบกั้น |
คง BarrierTakeMany.Options | waitForIncomplete (บูลีน waitForIncomplete) |
วิธีการสืบทอด
วิธีการสาธารณะ
สาธารณะคง BarrierTakeMany.Options allowSmallBatch (บูลีน allowSmallBatch)
พารามิเตอร์
allowSmallBatch | อนุญาตให้ส่งคืนน้อยกว่า num_elements รายการถ้าสิ่งกีดขวางถูกปิดแล้ว |
---|
สาธารณะคง BarrierTakeMany สร้าง ( ขอบเขต ขอบเขต Operand <String> จับ ตัวถูกดำเนินการ <Integer> numElements รายการ <ชั้น <? >> componentTypes, ตัวเลือก ... ตัวเลือก)
วิธีการจากโรงงานเพื่อสร้างคลาสที่ปิดการดำเนินการ BarrierTakeMany ใหม่
พารามิเตอร์
ขอบเขต | ขอบเขตปัจจุบัน |
---|---|
รับมือ | ที่จับกับสิ่งกีดขวาง |
numElements | เทนเซอร์องค์ประกอบเดียวที่มีจำนวนขององค์ประกอบที่จะใช้ |
ส่วนประกอบประเภท | ชนิดของแต่ละองค์ประกอบในค่า |
ตัวเลือก | ดำเนินการค่าแอตทริบิวต์ทางเลือก |
คืนสินค้า
- ตัวอย่างใหม่ของ BarrierTakeMany
ประชาชน เอาท์พุท <ยาว> ดัชนี ()
เทนเซอร์หนึ่งมิติของดัชนี มีความยาว num_elems ดัชนีเหล่านี้อ้างอิงถึงชุดงานที่มีการวางค่าไว้ในอุปสรรค (เริ่มต้นด้วย MIN_LONG และเพิ่มขึ้นตามแต่ละ BarrierInsertMany)
สาธารณะคง BarrierTakeMany.Options timeoutMs (Long timeoutMs)
พารามิเตอร์
หมดเวลาMs | หากคิวว่าง การดำเนินการนี้จะบล็อกไม่เกิน timeout_ms มิลลิวินาที หมายเหตุ: ยังไม่รองรับตัวเลือกนี้ |
---|
รายการสาธารณะ < เอาท์พุท <? >> ค่า ()
หนึ่งเทนเซอร์ทุกมิติต่อส่วนประกอบในองค์ประกอบกั้น ค่าทั้งหมดมีความยาว num_elements ในมิติที่ 0