lớp cuối cùng công khai BroadcastHelper
Toán tử trợ giúp để thực hiện các chương trình phát sóng kiểu XLA
Phát sóng `lhs` và `rhs` đến cùng một thứ hạng, bằng cách thêm kích thước 1 vào bất kỳ thứ nguyên nào trong số `lhs` và `rhs` có thứ hạng thấp hơn, sử dụng quy tắc phát sóng của XLA cho toán tử nhị phân.
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
| tĩnh <T mở rộng TType > BroadcastHelper <T> | |
| Đầu ra <T> | lhsĐầu ra () tensor LHS được phát sóng |
| Đầu ra <T> | rhsĐầu ra () tensor RHS được phát sóng |
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: "XlaBroadcastHelper"
Phương pháp công khai
public static BroadcastHelper <T> tạo ( Phạm vi phạm vi, Toán hạng <T> lhs, Toán hạng <T> rhs, Toán hạng <? mở rộng TNumber > BroadcastDims)
Phương thức xuất xưởng để tạo một lớp bao bọc một hoạt động BroadcastHelper mới.
Thông số
| phạm vi | phạm vi hiện tại |
|---|---|
| lh | tensor đầu vào LHS |
| ừ | tensor đầu vào RHS |
| phát sóngDims | đặc điểm kỹ thuật kích thước phát sóng kiểu XLA |
Trả lại
- một phiên bản mới của BroadcastHelper