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

org.tensorflow.op.RawOp
boolean cuối cùng
bằng (Đối tượng obj)
int cuối cùng
Mã Băm ()
Hoạt động
ồ ()
Trả về đơn vị tính toán này dưới dạng một Operation .
chuỗi cuối cùng
toString ()
boolean
bằng (Đối tượng arg0)
Lớp cuối cùng<?>
getClass ()
int
Mã Băm ()
khoảng trống cuối cùng
thông báo ()
khoảng trống cuối cùng
thông báoTất cả ()
Sợi dây
toString ()
khoảng trống cuối cùng
chờ đã (arg0 dài, int arg1)
khoảng trống cuối cùng
chờ đã (arg0 dài)
khoảng trống cuối cùng
Chờ đợi ()
org.tensorflow.op.Op
môi trường thực thi trừu tượng
env ()
Trả về môi trường thực thi mà op này được tạo trong đó.
hoạt động trừu tượng
ồ ()
Trả về đơn vị tính toán này dưới dạng một Operation .
org.tensorflow.Operand
Đầu ra trừu tượng <T>
asOutput ()
Trả về tay cầm tượng trưng của tensor.
trừu tượng T
asTensor ()
Trả về tensor ở toán hạng này.
hình dạng trừu tượng
hình dạng ()
Trả về hình dạng (có thể đã biết một phần) của tensor được tham chiếu bởi Output của toán hạng này.
Lớp trừu tượng<T>
kiểu ()
Trả về kiểu tensor của toán hạng này
org.tensorflow.ndarray.Shaped
int trừu tượng
thứ hạng ()
hình dạng trừu tượng
hình dạng ()
trừu tượng dài
kích cỡ ()
Tính toán và trả về tổng kích thước của vùng chứa này, theo số lượng giá trị.

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.