パブリック最終クラスSparseMatMul
行列「a」と行列「b」を掛けます。
入力は 2 次元行列である必要があり、「a」の内部次元は「b」の外部次元と一致する必要があります。 "a" と "b" は両方とも、SparseTensor ではなく Tensor でなければなりません。この演算は、「a」または「b」の少なくとも 1 つがスパースである場合、つまりゼロ値の割合が大きいという意味で最適化されています。 1 つのプラットフォームでこれを使用した場合と密行列の乗算を使用した場合の損益分岐点は、疎行列の 30% がゼロ値でした。
この操作の勾配計算では、入力勾配が Relu からのものである場合にのみ、入力勾配のスパース性が利用されます。
ネストされたクラス
| クラス | SparseMatMul.オプション | SparseMatMulのオプションの属性 | |
定数
| 弦 | OP_NAME | TensorFlow コア エンジンによって認識される、この演算の名前 |
パブリックメソッド
| 静的SparseMatMul.Options | aIsSparse (ブール値 aIsSparse) |
| 出力< TFloat32 > | asOutput () テンソルのシンボリック ハンドルを返します。 |
| 静的SparseMatMul.Options | bIsSparse (ブール値 bIsSparse) |
| 静的SparseMatMul | |
| 出力< TFloat32 > | 製品（） |
| 静的SparseMatMul.Options | transposeA (ブールtransposeA) |
| 静的SparseMatMul.Options | transposeB (ブールtransposeB) |
継承されたメソッド
定数
パブリック静的最終文字列OP_NAME
TensorFlow コア エンジンによって認識される、この演算の名前
定数値: "SparseMatMul"
パブリックメソッド
public Output < TFloat32 > asOutput ()
テンソルのシンボリック ハンドルを返します。
TensorFlow オペレーションへの入力は、別の TensorFlow オペレーションの出力です。このメソッドは、入力の計算を表すシンボリック ハンドルを取得するために使用されます。
public static SparseMatMul create (スコープscope、オペランド<? extends TNumber > a、オペランド<? extends TNumber > b、オプション... options)
新しい SparseMatMul オペレーションをラップするクラスを作成するファクトリ メソッド。
パラメーター
| 範囲 | 現在のスコープ |
|---|---|
| オプション | オプションの属性値を持ちます |
戻り値
- SparseMatMul の新しいインスタンス