Lấy số lượng phần tử đã hoàn thành nhất định từ một rào cản.
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.
Các lớp lồng nhau
lớp học | BarrierTakeMany.Options | Thuộc tính tùy chọn cho BarrierTakeMany |
Hằng số
Sợi dây | OP_NAME | Tên của op này, được biết đến bởi công cụ lõi TensorFlow |
Phương pháp công khai
BarrierTakeMany.Options tĩnh | allowSmallBatch (Boolean allowSmallBatch) |
rào cản tĩnh TakeMany | |
Đầu ra < TInt64 > | chỉ số () Tensor một chiều của các chỉ số, có độ dài num_elems. |
Đầu ra < TString > | phím () Một tenxơ một chiều của các phím, có độ dài là num_elements. |
BarrierTakeMany.Options tĩnh | timeoutMs (Thời gian chờ dàiMs) |
Danh sách< Đầu ra <?>> | giá trị () Một tenxơ chiều bất kỳ cho mỗi thành phần trong phần tử rào cản. |
BarrierTakeMany.Options tĩnh | waitForIncomplete (Boolean waitForIncomplete) |
Phương pháp kế thừa
Hằng số
Chuỗi cuối cùng tĩnh công khai OP_NAME
Tên của op này, được biết đến bởi công cụ lõi TensorFlow
Phương pháp công khai
tĩnh công khai BarrierTakeMany.Options allowSmallBatch (Boolean allowSmallBatch)
Thông số
cho phépSmallBatch | Cho phép trả lại ít hơn num_elements mục nếu rào cản đã được đóng. |
---|
tĩnh công khai BarrierTakeMany tạo ( Phạm vi phạm vi , Toán hạng < TString > xử lý, Toán hạng < TInt32 > numElements, Danh sách <Class<? mở rộng TType >> loại thành phần, Tùy chọn... tùy chọn)
Phương thức xuất xưởng để tạo một lớp bao bọc một hoạt động BarrierTakeMany mới.
Thông số
phạm vi | phạm vi hiện tại |
---|---|
xử lý | Tay cầm cho một rào cản. |
số phần tử | Một tenxơ một phần tử chứa số phần tử cần lấy. |
các loại thành phần | Loại của mỗi thành phần trong một giá trị. |
tùy chọn | mang các giá trị thuộc tính tùy chọn |
Trả lại
- một phiên bản mới của BarrierTakeMany
Đầu ra công khai < TInt64 > chỉ số ()
Tensor 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 rào cản (bắt đầu bằng MIN_LONG và tăng dần với mỗi BarrierInsertMany).
public static BarrierTakeMany.Options timeoutMs (Thời gian chờ dàiMs)
Thông số
hết thời gian chờ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ợ. |
---|
Danh sách công khai< Kết quả <?>> giá trị ()
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.