テンソルフロー::作戦::ママル
#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」は乗算の前に転置されます。
戻り値:
-
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
演算子::tensorflow::入力
operator::tensorflow::Input() const
演算子::tensorflow::出力
operator::tensorflow::Output() const
パブリック静的関数
トランスポーズA
Attrs TransposeA( bool x )
トランスポーズB
Attrs TransposeB( bool x )