MatMul

パブリック最終クラス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>
create (スコープscope、オペランド<T> a、オペランド<T> b、オプション...オプション)
新しい MatMul オペレーションをラップするクラスを作成するファクトリ メソッド。
出力<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出力<T> product ()

public static MatMul.Options transposeA (Boolean transposeA)

パラメーター
トランスポーズA true の場合、「a」は乗算の前に転置されます。

public static MatMul.Options transposeB (Boolean transposeB)

パラメーター
トランスポーズB true の場合、「b」は乗算の前に転置されます。