CollectivePermute

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 trường hợp 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]`.

Phương pháp công cộng

Đầu ra <T>
asOutput ()
Trả về phần điều khiển tượng trưng của một tenxơ.
tĩnh <T> CollectivePermute <T>
tạo (Phạm vi phạm vi , đầu vào Toán hạng <T>, Toán hạng <Số nguyên> nguồnTargetPairs)
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.
Đầu ra <T>
đầu ra ()
Đầu vào được phép.

Phương pháp kế thừa

Phương pháp công cộng

Đầu ra công khai <T> asOutput ()

Trả về phần điều khiển tượng trưng của một tenxơ.

Đầ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 tử <Integer> 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ồnTargetPairs 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

Đầu ra công khai đầu ra <T> ()

Đầu vào được phép.