Giải hệ phương trình tam giác.
Giải hệ phương trình tam giác. Hỗ trợ kích thước lô và nhiều bên phải cho mỗi bên trái. Trên CPU, giải pháp được tính toán thông qua việc loại bỏ Gaussian có hoặc không có xoay vòng một phần, tùy thuộc vào thuộc tính `partial_pivoting`. Trên GPU, thư viện cuSPARSE của Nvidia được sử dụng: https://docs.nvidia.com/cuda/cusparse/index.html#gtsv Việc xoay vòng một phần chưa được chương trình phụ trợ XLA hỗ trợ.
Các lớp lồng nhau
| lớp học | TridiagonalSolve.Options | Thuộc tính tùy chọn cho TridiagonalSolve | |
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 > TridiagonalSolve <T> | tạo ( Phạm vi phạm vi, Đường chéo toán hạng <T>, Toán hạng <T> rhs, Tùy chọn... tùy chọn) Phương thức xuất xưởng để tạo một lớp bao bọc một thao tác TridiagonalSolve mới. |
| Đầu ra <T> | đầu ra () Tenxor hình `[..., M, K]` chứa nghiệm |
| TridiagonalSolve.Options tĩnh | Xoay một phần (Boolean một phầnPivoting) |
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
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 TridiagonalSolve <T> tạo ( Phạm vi phạm vi, Đường chéo toán hạng <T>, Toán hạng <T> rhs, Tùy chọn... tùy chọn)
Phương thức xuất xưởng để tạo một lớp bao bọc một thao tác TridiagonalSolve mới.
Thông số
| phạm vi | phạm vi hiện tại |
|---|---|
| đường chéo | Tenxor hình `[..., 3, M]` có 2 chiều trong cùng biểu thị các ma trận tam giác với 3 hàng là siêu chéo, đường chéo và đường chéo theo thứ tự. Phần tử cuối cùng của đường chéo và phần tử đầu tiên của đường chéo phụ bị bỏ qua. |
| ừ | Tenxor có hình dạng `[..., M, K]`, biểu thị K cạnh phải trên mỗi cạnh trái. |
| tùy chọn | mang các giá trị thuộc tính tùy chọn |
Trả lại
- một phiên bản mới của TridiagonalSolve
public static TridiagonalSolve.Options một phầnPivoting (Boolean một phầnPivoting)
Thông số
| xoay một phần | Có áp dụng xoay vòng một phần hay không. Xoay một phần làm cho quy trình ổn định hơn nhưng chậm hơn. |
|---|