TridiagonalSolve

TridiagonalSolve kelas akhir publik

Memecahkan sistem persamaan tridiagonal.

Memecahkan sistem persamaan tridiagonal. Mendukung dimensi batch dan beberapa sisi kanan untuk setiap sisi kiri. Pada CPU, solusi dihitung melalui eliminasi Gaussian dengan atau tanpa pivoting parsial, bergantung pada atribut `partial_pivoting`. Pada GPU, pustaka cuSPARSE Nvidia digunakan: https://docs.nvidia.com/cuda/cusparse/index.html#gtsv Pivot parsial belum didukung oleh backend XLA.

Kelas Bersarang

kelas Pemecahan Tridiagonal.Opsi Atribut opsional untuk TridiagonalSolve

Metode Publik

Keluaran <T>
sebagai Keluaran ()
Mengembalikan pegangan simbolik tensor.
statis <T> Pemecahan Tridiagonal <T>
buat ( Lingkup lingkup, Operand <T> diagonal, Operand <T> rhs, Pilihan... pilihan)
Metode pabrik untuk membuat kelas yang membungkus operasi TridiagonalSolve baru.
Keluaran <T>
keluaran ()
Tensor berbentuk `[..., M, K]` yang berisi solusi
TridiagonalSolve.Options statis
parsialPivoting (Pivoting parsial Boolean)
TridiagonalSolve.Options statis
perturbSingular (Boolean perturbSingular)

Metode Warisan

Metode Publik

Keluaran publik <T> sebagai Keluaran ()

Mengembalikan pegangan simbolik tensor.

Masukan ke operasi TensorFlow adalah keluaran dari operasi TensorFlow lainnya. Metode ini digunakan untuk mendapatkan pegangan simbolis yang mewakili perhitungan input.

public static TridiagonalSolve <T> buat ( Lingkup lingkup, Operan <T> diagonal, Operan <T> rhs, Opsi... opsi)

Metode pabrik untuk membuat kelas yang membungkus operasi TridiagonalSolve baru.

Parameter
cakupan ruang lingkup saat ini
diagonal Tensor bentuk `[..., 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

Keluaran publik <T> keluaran ()

Tensor berbentuk `[..., M, K]` yang berisi solusi

TridiagonalSolve.Options parsialPivoting statis publik (Pivoting parsial Boolean)

Parameter
parsialberputar Apakah akan menerapkan perputaran parsial. Perputaran parsial membuat prosedur lebih stabil, namun lebih lambat.

public static TridiagonalSolve.Options perturbSingular (Boolean perturbSingular)