テンソルフロー::作戦::バッチマットMulV2

#include <math_ops.h>

2 つのテンソルのスライスをバッチで乗算します。

まとめ

Tensor xyのすべてのスライスを乗算し (各スライスはバッチの要素として表示できます)、個々の結果を同じバッチ サイズの単一の出力テンソルに配置します。個々のスライスはオプションで、乗算の前にadj_xまたはadj_yフラグをTrue (デフォルトではFalse ) に設定することにより、隣接させることができます (行列を隣接させるとは、行列を転置および共役することを意味します)。

入力テンソルxおよびy 、形状[..., r_x, c_x]および[..., r_y, c_y]を持つ 2 次元以上です。

出力テンソルは、形状[..., r_o, c_o]の 2 次元以上です。ここで、

r_o = c_x if adj_x else r_x
c_o = r_y if adj_y else c_y

output[..., :, :] = matrix(x[..., :, :]) * matrix(y[..., :, :])
として計算されます。

: BatchMatMulV2バッチ次元でのブロードキャストをサポートしています。ブロードキャストの詳細については、こちらをご覧ください

引数:

  • スコープ:スコープオブジェクト
  • x: 形状[..., r_x, c_x]の 2 次元以上。
  • y: 形状[..., r_y, c_y]の 2 次元以上。

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

  • adj_x: Trueの場合、 xのスライスを隣接させます。デフォルトはFalseです。
  • adj_y: Trueの場合、 yのスライスを隣接させます。デフォルトはFalseです。

戻り値:

  • Output : 形状[..., r_o, c_o]の 3 次元以上

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

BatchMatMulV2 (const :: tensorflow::Scope & scope, :: tensorflow::Input x, :: tensorflow::Input y)
BatchMatMulV2 (const :: tensorflow::Scope & scope, :: tensorflow::Input x, :: tensorflow::Input y, const BatchMatMulV2::Attrs & attrs)

パブリック属性

operation
output

公共機能

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

パブリック静的関数

AdjX (bool x)
AdjY (bool x)

構造体

tensorflow:: ops:: BatchMatMulV2:: Attrs

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

パブリック属性

手術

Operation operation

出力

::tensorflow::Output output

公共機能

バッチマットMulV2

 BatchMatMulV2(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input x,
  ::tensorflow::Input y
)

バッチマットMulV2

 BatchMatMulV2(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input x,
  ::tensorflow::Input y,
  const BatchMatMulV2::Attrs & attrs
)

ノード

::tensorflow::Node * node() const 

演算子::tensorflow::入力

 operator::tensorflow::Input() const 

演算子::tensorflow::出力

 operator::tensorflow::Output() const 

パブリック静的関数

調整X

Attrs AdjX(
  bool x
)

調整

Attrs AdjY(
  bool x
)