テンソルフロー::作戦::ママル

#include <math_ops.h>

行列「a」と行列「b」を掛けます

まとめ

入力は 2 次元行列である必要があり、「a」の内部次元 (transpose_a が true の場合は転置後) は、「b」の外部次元 (transpose_b が true の場合は転置後) と一致する必要があります。

: GPU 上のMatMulのデフォルトのカーネル実装では、cublas を使用します。

引数:

オプションの属性 ( Attrsを参照):

  • transpose_a: true の場合、「a」は乗算の前に転置されます。
  • transpose_b: true の場合、「b」は乗算の前に転置されます。

戻り値:

コンストラクターとデストラクター

MatMul (const :: tensorflow::Scope & scope, :: tensorflow::Input a, :: tensorflow::Input b)
MatMul (const :: tensorflow::Scope & scope, :: tensorflow::Input a, :: tensorflow::Input b, const MatMul::Attrs & attrs)

パブリック属性

operation
product

公共機能

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

パブリック静的関数

TransposeA (bool x)
TransposeB (bool x)

構造体

tensorflow:: ops:: MatMul:: Attrs

MatMulのオプションの属性セッター。

パブリック属性

手術

Operation operation

製品

::tensorflow::Output product

公共機能

ママル

 MatMul(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input a,
  ::tensorflow::Input b
)

ママル

 MatMul(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input a,
  ::tensorflow::Input b,
  const MatMul::Attrs & attrs
)

ノード

::tensorflow::Node * node() const 

演算子::tensorflow::入力

 operator::tensorflow::Input() const 

演算子::tensorflow::出力

 operator::tensorflow::Output() const 

パブリック静的関数

トランスポーズA

Attrs TransposeA(
  bool x
)

トランスポーズB

Attrs TransposeB(
  bool x
)