NcclBroadcast

lớp cuối cùng công khai NcclBroadcast

Gửi `đầu vào` tới tất cả các thiết bị được kết nối với đầu ra.

Gửi `đầu vào` tới tất cả các thiết bị được kết nối với đầu ra.

Biểu đồ phải được xây dựng sao cho tất cả các op được kết nối với đầu ra đề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 của chương trình phát sóng. đầu ra: Giống như đầu vào. hình dạng: Hình dạng của tensor đầu vào.

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 > NcclBroadcast <T>
tạo (Phạm vi phạm vi , đầu vào Toán hạng <T>, Hình dạng)
Phương thức xuất xưởng để tạo một lớp bao bọc thao tác NcclBroadcast mới.
Đầu ra <T>

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: "NcclBroadcast"

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 NcclBroadcast <T> tạo (Phạm vi phạm vi , đầu vào Toán hạng <T>, Hình dạng hình dạng)

Phương thức xuất xưởng để tạo một lớp bao bọc thao tác NcclBroadcast 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 NcclBroadcast

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