tensorflow::ops::BarrierTakeMany

Lấy số lượng phần tử đã hoàn thành được chỉ định từ một rào chắn.

Bản tóm tắt

Thao tác này nối các tensor thành phần đã hoàn thiện dọc theo chiều thứ 0 để tạo thành một tensor thành phần duy nhất.

Các phần tử sẽ thoát ra khỏi rào chắn khi chúng hoàn thành và theo đúng thứ tự chúng được đưa vào rào chắn. Thông tin chỉ số cung cấp thông tin về lô mà mỗi phần tử ban đầu được đưa vào rào chắn.

Luận điểm:

• phạm vi: Một đối tượng Phạm vi
• tay cầm: Tay cầm của một rào chắn.
• num_elements: Một tensor đơn phần tử chứa số lượng phần tử cần lấy.
• component_types: Loại của từng thành phần trong một giá trị.

Các thuộc tính tùy chọn (xem Attrs ):

• allow_small_batch: Cho phép trả về ít hơn num_elements mục nếu rào chắn đã đóng.
• timeout_ms: Nếu hàng đợi trống, thao tác này sẽ bị chặn trong tối đa timeout_ms mili giây. Lưu ý: Tùy chọn này hiện chưa được hỗ trợ.

Trả về:

• Chỉ số Output : Một tensor chỉ số một chiều, có độ dài num_elems. Các chỉ số này tham chiếu đến lô mà các giá trị được đưa vào rào chắn (bắt đầu từ MIN_LONG và tăng dần với mỗi BarrierInsertMany ).
• Các khóa Output : Một tensor một chiều chứa các khóa, có độ dài là num_elements.
• Giá trị OutputList : Một tensor bất kỳ chiều cho mỗi thành phần trong một phần tử rào cản. Tất cả các giá trị đều có độ dài num_elements ở chiều thứ 0.

Thuộc tính công khai

::tensorflow::Output indices

::tensorflow::Output keys

Operation operation

::tensorflow::OutputList values

 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
)

Attrs AllowSmallBatch(
  bool x
)

Attrs TimeoutMs(
  int64 x
)

Attrs WaitForIncomplete(
  bool x
)