Memecahkan sistem persamaan tridiagonal.
Memecahkan sistem persamaan tridiagonal. Mendukung dimensi batch dan beberapa sisi kanan per setiap sisi kiri. Pada CPU, solusi dihitung melalui eliminasi Gaussian dengan atau tanpa pivot parsial, bergantung pada atribut `partial_pivoting`. Pada GPU, pustaka cuSPARSE Nvidia digunakan: https://docs.nvidia.com/cuda/cusparse/index.html#gtsv Pivot sebagian belum didukung oleh backend XLA.
Kelas Bersarang
kelas | TridiagonalSolve.Options | Atribut opsional untuk TridiagonalSolve |
Metode Publik
Output <T> | asOutput () Mengembalikan pegangan simbolis tensor. |
statis <T> TridiagonalSolve <T> | membuat ( Lingkup lingkup, Operan <T> diagonal, Operan <T> rhs, Options ... pilihan) Metode pabrik untuk membuat kelas yang membungkus operasi TridiagonalSolve baru. |
Output <T> | Output () Tensor berbentuk `[..., M, K]` yang berisi solusi |
statis TridiagonalSolve.Options | partialPivoting (Boolean partialPivoting) |
statis TridiagonalSolve.Options | perturbSingular (Boolean perturbSingular) |
Metode yang Diwarisi
Metode Publik
publik Keluaran <T> asOutput ()
Mengembalikan pegangan simbolis tensor.
Input ke operasi TensorFlow adalah output dari operasi TensorFlow lainnya. Metode ini digunakan untuk mendapatkan pegangan simbolis yang mewakili perhitungan input.
public static TridiagonalSolve <T> membuat ( Lingkup lingkup, Operan <T> diagonal, Operan <T> rhs, Options ... pilihan)
Metode pabrik untuk membuat kelas yang membungkus operasi TridiagonalSolve baru.
Parameter
cakupan | lingkup saat ini |
---|---|
diagonal | Tensor berbentuk `[..., 3, M]` yang 2 dimensi terdalamnya mewakili matriks tridiagonal dengan tiga baris berturut-turut adalah superdiagonal, diagonal, dan subdiagonal. Elemen terakhir dari superdiagonal dan elemen pertama dari subdiagonal diabaikan. |
rhs | Tensor berbentuk `[..., M, K]`, mewakili K sisi kanan per setiap sisi kiri. |
pilihan | membawa nilai atribut opsional |
Kembali
- contoh baru dari TridiagonalSolve
public static TridiagonalSolve.Options partialPivoting (Boolean partialPivoting)
Parameter
sebagianPivoting | Apakah akan menerapkan pivot parsial. Pemutaran sebagian membuat prosedur lebih stabil, tetapi lebih lambat. |
---|