日付を保存! Google I / Oが5月18日から20日に戻ってきます今すぐ登録
このページは Cloud Translation API によって翻訳されました。
Switch to English

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

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

MatMul

 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
)