Peringatan: API ini tidak digunakan lagi dan akan dihapus di versi TensorFlow mendatang setelah penggantinya stabil.

TridiagonalSolve

public class akhir TridiagonalSolve

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

publik Keluaran <T> Output ()

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

public static TridiagonalSolve.Options partialPivoting (Boolean partialPivoting)

Parameter
sebagianPivoting Apakah akan menerapkan pivot parsial. Pemutaran sebagian membuat prosedur lebih stabil, tetapi lebih lambat.

public static TridiagonalSolve.Options perturbSingular (Boolean perturbSingular)