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

入力に対して最大プーリングを実行し、最大値とインデックスの両方を出力します。

まとめ

argmaxのインデックスは平坦化されるため、位置[b, y, x, c]の最大値は平坦化されたインデックスになります。 include_batch_in_indexが False の場合は、 (y * width + x) * channels + c 。 ((b * height + y) * width + x) * channels + c include_batch_in_indexが True の場合。

返されるインデックスは、パディングが含まれており、数学的に正しい答えが外側 (負であるか大きすぎる) であっても、平坦化する前は常に[0, height) x [0, width)になります。これはバグですが、特にフラット化のため、安全な下位互換性のある方法で修正することは困難です。

引数:

• スコープ:スコープオブジェクト
• 入力: 形状[batch, height, width, channels]の 4-D。プールする入力。
• ksize: 入力テンソルの各次元のウィンドウのサイズ。
• strides: 入力テンソルの各次元のスライディング ウィンドウのストライド。
• padding: 使用するパディング アルゴリズムのタイプ。

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

• include_batch_in_index: argmaxのフラット化されたインデックスにバッチ ディメンションを含めるかどうか。

戻り値:

• 出力Output : プールされた最大出力テンソル。
• Output最大: 4-D。各出力に対して選択された最大値の平坦化されたインデックス。

パブリック属性

::tensorflow::Output argmax

Operation operation

::tensorflow::Output output

 MaxPoolWithArgmax(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  const gtl::ArraySlice< int > & ksize,
  const gtl::ArraySlice< int > & strides,
  StringPiece padding
)

 MaxPoolWithArgmax(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  const gtl::ArraySlice< int > & ksize,
  const gtl::ArraySlice< int > & strides,
  StringPiece padding,
  const MaxPoolWithArgmax::Attrs & attrs
)

Attrs IncludeBatchInIndex(
  bool x
)

Attrs Targmax(
  DataType x
)