Giảm `đầu vào` từ `num_devices` bằng cách sử dụng `reduction` xuống một thiết bị duy nhất.
Giảm `đầu vào` từ `num_devices` bằng cách sử dụng `reduction` xuống một thiết bị duy nhất.
Biểu đồ phải được xây dựng sao cho tất cả các đầu vào đều được gán thiết bị hợp lệ và bản thân op được gán một trong các thiết bị này.
đầu vào: Đầu vào để giảm. dữ liệu: giá trị mức giảm trên tất cả các thiết bị `num_devices`. giảm: hoạt động giảm để thực hiện.
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
| Đầu ra <T> | asOutput () Trả về tay cầm tượng trưng của tensor. |
| tĩnh <T mở rộng TNumber > NcclReduce <T> | tạo ( Phạm vi phạm vi, đầu vào Iterable< Operand <T>>, Giảm chuỗi) Phương thức xuất xưởng để tạo một lớp bao bọc thao tác NcclReduce mới. |
| Đầu ra <T> | dữ liệu () |
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
Đầu ra công khai <T> asOutput ()
Trả về tay cầm tượng trưng của tensor.
Đầu vào của các hoạt động TensorFlow là đầu ra của một hoạt động TensorFlow khác. Phương pháp này được sử dụng để thu được một thẻ điều khiển mang tính biểu tượng đại diện cho việc tính toán đầu vào.
public static NcclReduce <T> tạo (Phạm vi phạm vi , đầu vào Iterable< Operand <T>>, Giảm chuỗi)
Phương thức xuất xưởng để tạo một lớp bao bọc thao tác NcclReduce mới.
Thông số
| phạm vi | phạm vi hiện tại |
|---|
Trả lại
- một phiên bản mới của NcclReduce