tensorflow:: ops:: MatMul

#include <math_ops.h>

行列「a」に行列「b」を乗算します。

まとめ

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

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

議論:

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

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

返品:

  • Output :積テンソル。

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

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 

operator::tensorflow::Input

 operator::tensorflow::Input() const 

operator::tensorflow::Output

 operator::tensorflow::Output() const 

パブリック静的関数

TransposeA

Attrs TransposeA(
  bool x
)

TransposeB

Attrs TransposeB(
  bool x
)