เทนเซอร์โฟลว์ :: ops :: BarrierTakeMany

#include <data_flow_ops.h>

รับองค์ประกอบที่เสร็จสมบูรณ์ตามจำนวนที่กำหนดจากอุปสรรค

สรุป

การดำเนินการนี้เชื่อมต่อเทนเซอร์ขององค์ประกอบที่เสร็จสมบูรณ์ตามมิติที่ 0 เพื่อสร้างเทนเซอร์ของส่วนประกอบเดียว

องค์ประกอบจะออกมาจากสิ่งกีดขวางเมื่อเสร็จสมบูรณ์และตามลำดับที่วางลงในสิ่งกีดขวาง เอาต์พุตดัชนีจะให้ข้อมูลเกี่ยวกับแบตช์ที่แต่ละองค์ประกอบถูกแทรกลงในอุปสรรค

อาร์กิวเมนต์:

  • ขอบเขต: วัตถุ ขอบเขต
  • ที่จับ: ที่จับกับสิ่งกีดขวาง
  • num_elements: เทนเซอร์องค์ประกอบเดียวที่มีจำนวนองค์ประกอบที่จะรับ
  • component_types: ประเภทของแต่ละองค์ประกอบในค่า

แอตทริบิวต์เสริม (ดู Attrs ):

  • allow_small_batch: อนุญาตให้ส่งคืนไอเท็มน้อยกว่า num_elements หากสิ่งกีดขวางถูกปิดไปแล้ว
  • timeout_ms: หากคิวว่างการดำเนินการนี้จะบล็อกได้นานถึง timeout_ms มิลลิวินาที หมายเหตุ: ยังไม่รองรับตัวเลือกนี้

ผลตอบแทน:

  • ดัชนี Output : เทนเซอร์หนึ่งมิติของดัชนีที่มีความยาว num_elems ดัชนีเหล่านี้หมายถึงชุดที่มีการวางค่าลงในอุปสรรค (เริ่มต้นด้วย MIN_LONG และเพิ่มขึ้นตามแต่ละ BarrierInsertMany )
  • คีย์ Output : คีย์เทนเซอร์หนึ่งมิติที่มีตัวเลขความยาว
  • ค่า OutputList : หนึ่งมิติเทนเซอร์ต่อส่วนประกอบในองค์ประกอบกั้น ค่า ทั้งหมด มีความยาว num_elements ในมิติที่ 0

ผู้สร้างและผู้ทำลาย

BarrierTakeMany (const :: tensorflow::Scope & scope, :: tensorflow::Input handle, :: tensorflow::Input num_elements, const DataTypeSlice & component_types)
BarrierTakeMany (const :: tensorflow::Scope & scope, :: tensorflow::Input handle, :: tensorflow::Input num_elements, const DataTypeSlice & component_types, const BarrierTakeMany::Attrs & attrs)

คุณลักษณะสาธารณะ

indices
keys
operation
values

ฟังก์ชั่นคงที่สาธารณะ

AllowSmallBatch (bool x)
TimeoutMs (int64 x)
WaitForIncomplete (bool x)

โครงสร้าง

tensorflow :: ops :: BarrierTakeMany :: Attrs

ตัวตั้งค่าแอตทริบิวต์เสริมสำหรับ BarrierTakeMany

คุณลักษณะสาธารณะ

ดัชนี

::tensorflow::Output indices

กุญแจ

::tensorflow::Output keys

การดำเนินการ

Operation operation

ค่า

::tensorflow::OutputList values

หน้าที่สาธารณะ

BarrierTakeMany

 BarrierTakeMany(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input handle,
  ::tensorflow::Input num_elements,
  const DataTypeSlice & component_types
)

BarrierTakeMany

 BarrierTakeMany(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input handle,
  ::tensorflow::Input num_elements,
  const DataTypeSlice & component_types,
  const BarrierTakeMany::Attrs & attrs
)

ฟังก์ชั่นคงที่สาธารณะ

AllowSmallBatch

Attrs AllowSmallBatch(
  bool x
)

หมดเวลา

Attrs TimeoutMs(
  int64 x
)

WaitForIncomplete

Attrs WaitForIncomplete(
  bool x
)