org.tensorflow.op.distribute

Các lớp học

NcclAllReduce <T mở rộng TNumber > Xuất ra một tensor chứa mức giảm trên tất cả các tensor đầu vào.
NcclBroadcast <T mở rộng TNumber > Gửi `đầu vào` tới tất cả các thiết bị được kết nối với đầu ra.
NcclReduce <T mở rộng TNumber > 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.