lớp học cuối cùng công khai Einsum
Một op hỗ trợ op einsum cơ bản với 2 đầu vào và 1 đầu ra.
Op này có hiệu suất TPU tốt hơn vì nó không có các hoạt động định hình lại và chuyển vị rõ ràng như tf.einsum.
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 > Einsum <T> | |
| Đầu ra <T> | sản phẩm () |
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: "XlaEinsum"
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 Einsum <T> tạo ( Phạm vi phạm vi, Toán hạng <T> a, Toán hạng <T> b, Phương trình chuỗi)
Phương thức xuất xưởng để tạo một lớp bao bọc một hoạt động Einsum 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 Einsum