dòng chảy :: hoạt động :: BarrierTakeMany

#include <data_flow_ops.h>

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

Bản tóm tắt

Phép toán này nối các tensor thành phần đã hoàn thành với kích thước thứ 0 để tạo ra một tensor thành phần duy nhất.

Các phần tử ra khỏi kết giới khi chúng hoàn thành và theo thứ tự chúng được đặt vào kết giới. Đầ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 rào cản.

Tranh luận:

  • phạm vi: Một đối tượng Phạm vi
  • tay cầm: Tay cầm để chắn.
  • num_elements: Một tenxơ đơn phần tử chứa số phần tử cần lấy.
  • component_types: Kiểu của mỗi 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ả 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, hoạt động 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ợ.

Lợi nhuận:

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

Người xây dựng và Người phá hủy

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)

Thuộc tính công cộng

indices
keys
operation
values

Các chức năng tĩnh công khai

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

Cấu trúc

tensorflow :: ops :: BarrierTakeMany :: Attrs

Bộ thiết lập thuộc tính tùy chọn cho BarrierTakeMany .

Thuộc tính công cộng

chỉ số

::tensorflow::Output indices

chìa khóa

::tensorflow::Output keys

hoạt động

Operation operation

giá trị

::tensorflow::OutputList values

Chức năng công cộng

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
)

Các chức năng tĩnh công khai

AllowSmallBatch

Attrs AllowSmallBatch(
  bool x
)

TimeoutMs

Attrs TimeoutMs(
  int64 x
)

WaitForIncomplete

Attrs WaitForIncomplete(
  bool x
)