TridiagonalSolve

パブリック最終クラスTridiagonalSolve

三重対角方程式系を解きます。

三重対角方程式系を解きます。バッチディメンションと各左側ごとに複数の右側をサポートします。 CPU では、「partial_pivoting」属性に応じて、部分ピボットの有無にかかわらず、ガウス消去法によって解が計算されます。 GPU では、Nvidia の cuSPARSE ライブラリが使用されます: https://docs.nvidia.com/cuda/cusparse/index.html#gtsv 部分ピボットは、XLA バックエンドではまだサポートされていません。

ネストされたクラス

クラスTridiagonalSolve.オプションTridiagonalSolveのオプションの属性

パブリックメソッド

出力<T>
asOutput ()
テンソルのシンボリック ハンドルを返します。
静的 <T> TridiagonalSolve <T>
create (スコープscope、オペランド<T>対角線、オペランド<T>右辺、オプション...オプション)
新しい TridiagonalSolve 操作をラップするクラスを作成するファクトリ メソッド。
出力<T>
出力()
解を含む形状 `[..., M, K]` のテンソル
静的TridiagonalSolve.Options
部分ピボット(ブール型の部分ピボット)

継承されたメソッド

パブリックメソッド

public Output <T> asOutput ()

テンソルのシンボリック ハンドルを返します。

TensorFlow オペレーションへの入力は、別の TensorFlow オペレーションの出力です。このメソッドは、入力の計算を表すシンボリック ハンドルを取得するために使用されます。

public static TridiagonalSolve <T> create (スコープスコープ、オペランド<T> 対角線、オペランド<T> rhs、オプション...オプション)

新しい TridiagonalSolve 操作をラップするクラスを作成するファクトリ メソッド。

パラメーター
範囲現在のスコープ
対角線形状 `[..., 3, M]` のテンソル。その最も内側の 2 次元は、3 行が順に上対角、対角、下対角である三重対角行列を表します。上対角の最後の要素と下対角の最初の要素は無視されます。
右肩各左辺ごとに K 個の右辺を表す、形状 `[..., M, K]` のテンソル。
オプションオプションの属性値を持ちます
戻り値
  • TridiagonalSolve の新しいインスタンス

public出力<T>出力()

解を含む形状 `[..., M, K]` のテンソル

public static TridiagonalSolve.OptionspartialPivoting ( BooleanpartialPivoting)

パラメーター
部分ピボット部分ピボットを適用するかどうか。部分ピボットにより手順はより安定しますが、速度は遅くなります。