TridiagonalSolve

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

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

Giá trị không đổi: "Giải tam giác"

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

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

Tenxor hình `[..., M, K]` chứa nghiệm

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.