dòng chảy căng:: ôi:: Rào CảnLấyNhiều

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

Bản tóm tắt

Hoạt động này nối các tensor thành phần phần tử hoàn chỉnh dọc theo chiều thứ 0 để tạo thành một tensor thành phần đơn.

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

Lập luận:

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

Thuộc tính tùy chọn (xem Attrs ):

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

Trả về:

• Các chỉ số Output : Một tenxơ một chiều của các chỉ số, có độ dài num_elems. Các chỉ số này đề cập đến lô trong đó các giá trị được đặt vào hàng rào (bắt đầu bằng MIN_LONG và tăng dần theo mỗi BarrierInsertMany ).
• Các phím Output : Một tensor một chiều của các phím, có độ dài là num_elements.
• Các giá trị OutputList : Một tenxơ chiều bất kỳ cho mỗi thành phần trong 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
)