lớp cuối cùng công khai AllReduce
Giảm lẫn nhau nhiều tensor có cùng loại và hình dạng.
Các lớp lồng nhau
| lớp học | AllReduce.Options | Thuộc tính tùy chọn cho AllReduce | |
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 AllReduce.Options | CommunicationsHint (Giao tiếp chuỗiHint) |
| tĩnh <T mở rộng TNumber > AllReduce <T> | tạo ( Phạm vi phạm vi, đầu vào Toán hạng <T>, Kích thước nhóm dài, Khóa nhóm dài, Khóa trường hợp dài, Chuỗi hợp nhấtOp, Chuỗi cuối cùng, Danh sách <Long> subdivOffsets, 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 thao tác AllReduce mới. |
| Đầu ra <T> | dữ liệu () |
| tĩnh AllReduce.Options | hết thời gian chờGiây (Thời gian chờ nổiGiây) |
| tĩnh AllReduce.Options | chờFor (Danh sách <Dài> chờ) |
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
Giá trị không đổi: "CollectiveReduce"
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 AllReduce <T> tạo ( Phạm vi phạm vi , đầu vào Toán hạng <T>, Long groupSize, Long groupKey, Long instanceKey, String mergeOp, String FinalOp, List<Long> subdivOffsets, Options... tùy chọn)
Phương thức xuất xưởng để tạo một lớp bao bọc một thao tác AllReduce mới.
Thông số
| phạm vi | phạm vi hiện tại |
|---|---|
| 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 AllReduce