パブリック最終クラスTridiagonalSolve
三重対角方程式系を解きます。
三重対角方程式系を解きます。バッチディメンションと各左側ごとに複数の右側をサポートします。 CPU では、「partial_pivoting」属性に応じて、部分ピボットの有無にかかわらず、ガウス消去法によって解が計算されます。 GPU では、Nvidia の cuSPARSE ライブラリが使用されます: https://docs.nvidia.com/cuda/cusparse/index.html#gtsv 部分ピボットは、XLA バックエンドではまだサポートされていません。
ネストされたクラス
クラス | TridiagonalSolve.オプション | TridiagonalSolve のオプションの属性 |
パブリックメソッド
出力<T> | asOutput () テンソルのシンボリック ハンドルを返します。 |
静的 <T> TridiagonalSolve <T> | |
出力<T> | 出力() 解を含む形状 `[..., M, K]` のテンソル |
静的TridiagonalSolve.Options | 部分ピボット(ブール型の部分ピボット) |
静的TridiagonalSolve.Options | perturbSingular (ブール値の perturbSingular) |
継承されたメソッド
パブリックメソッド
public Output <T> asOutput ()
テンソルのシンボリック ハンドルを返します。
TensorFlow オペレーションへの入力は、別の TensorFlow オペレーションの出力です。このメソッドは、入力の計算を表すシンボリック ハンドルを取得するために使用されます。
public static TridiagonalSolve <T> create (スコープスコープ、オペランド<T> 対角線、オペランド<T> rhs、オプション...オプション)
新しい TridiagonalSolve 操作をラップするクラスを作成するファクトリ メソッド。
パラメーター
範囲 | 現在のスコープ |
---|---|
対角線 | 形状 `[..., 3, M]` のテンソル。その最も内側の 2 次元は、3 行が順に上対角、対角、下対角である三重対角行列を表します。上対角の最後の要素と下対角の最初の要素は無視されます。 |
右肩 | 各左辺ごとに K 個の右辺を表す、形状 `[..., M, K]` のテンソル。 |
オプション | オプションの属性値を持ちます |
戻り値
- TridiagonalSolve の新しいインスタンス
public static TridiagonalSolve.OptionspartialPivoting ( BooleanpartialPivoting)
パラメーター
部分ピボット | 部分ピボットを適用するかどうか。部分ピボットにより手順はより安定しますが、速度は遅くなります。 |
---|