パブリック最終クラスMatMul
行列「a」と行列「b」を掛けます。
入力は 2 次元行列である必要があり、「a」の内部次元 (transpose_a が true の場合は転置後) は、「b」の外部次元 (transpose_b が true の場合は転置後) と一致する必要があります。
注: GPU 上の MatMul のデフォルトのカーネル実装では、cublas を使用します。
ネストされたクラス
クラス | MatMul.オプション | MatMul のオプションの属性 |
定数
弦 | OP_NAME | TensorFlow コア エンジンによって認識される、この演算の名前 |
パブリックメソッド
出力<T> | asOutput () テンソルのシンボリック ハンドルを返します。 |
static <T extends TType > MatMul <T> | |
出力<T> | 製品() |
静的MatMul.Options | transposeA (ブールtransposeA) |
静的MatMul.Options | transposeB (ブールtransposeB) |
継承されたメソッド
定数
パブリック静的最終文字列OP_NAME
TensorFlow コア エンジンによって認識される、この演算の名前
定数値: "MatMul"
パブリックメソッド
public Output <T> asOutput ()
テンソルのシンボリック ハンドルを返します。
TensorFlow オペレーションへの入力は、別の TensorFlow オペレーションの出力です。このメソッドは、入力の計算を表すシンボリック ハンドルを取得するために使用されます。
public static MatMul <T> create (スコープscope、オペランド<T> a、オペランド<T> b、オプション...オプション)
新しい MatMul オペレーションをラップするクラスを作成するファクトリ メソッド。
パラメーター
範囲 | 現在のスコープ |
---|---|
オプション | オプションの属性値を持ちます |
戻り値
- MatMul の新しいインスタンス
public static MatMul.Options transposeA (Boolean transposeA)
パラメーター
トランスポーズA | true の場合、「a」は乗算の前に転置されます。 |
---|
public static MatMul.Options transposeB (Boolean transposeB)
パラメーター
トランスポーズB | true の場合、「b」は乗算の前に転置されます。 |
---|