lớp cuối cùng công khai CollectivePermute
Một giải pháp để hoán vị các tensor trên các phiên bản TPU được sao chép.
Mỗi phiên bản cung cấp đầu vào riêng của nó.
Ví dụ: giả sử có 4 phiên bản TPU: `[A, B, C, D]`. Truyền source_target_pairs=`[[0,1],[1,2],[2,3],[3,0]]` nhận được kết quả đầu ra: `[D, A, B, C]`.
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 TType > CollectivePermute <T> | |
| Đầu ra <T> | đầu ra () Đầu vào được cho phép. |
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: "CollectivePermute"
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 CollectivePermute <T> tạo (Phạm vi phạm vi, đầu vào Toán hạng <T>, Toán hạng < TInt32 > sourceTargetPairs)
Phương thức xuất xưởng để tạo một lớp bao bọc một thao tác CollectivePermute mới.
Thông số
| phạm vi | phạm vi hiện tại |
|---|---|
| đầu vào | Đầu vào cục bộ được hoán vị. Hiện tại chỉ hỗ trợ float và bfloat16. |
| nguồnCặp mục tiêu | Một tensor có hình dạng [num_pairs, 2]. |
Trả lại
- một phiên bản mới của CollectivePermute