tensorflow::ops::MaxPoolWithArgmax

#include <nn_ops.h>

Esegue il max pooling sull'input e restituisce sia i valori massimi che gli indici.

Riepilogo

Gli indici in argmax vengono appiattiti, in modo che un valore massimo nella posizione [b, y, x, c] diventi indice appiattito: (y * width + x) * channels + c se include_batch_in_index è False; ((b * height + y) * width + x) * channels + c se include_batch_in_index è True.

Gli indici restituiti sono sempre compresi tra [0, height) x [0, width) prima dell'appiattimento, anche se è coinvolto il padding e la risposta matematicamente corretta è al di fuori di questo intervallo (o negativa o troppo grande). Questo è un bug, ma correggerlo in modo sicuro e retrocompatibile è difficile, soprattutto a causa dell'appiattimento.

Argomentazioni:

  • ambito: un oggetto Scope
  • Input: 4D con forma [batch, height, width, channels] . Input su cui eseguire il pooling.
  • ksize: La dimensione della finestra per ciascuna dimensione del tensore di input.
  • passi: Il passo della finestra scorrevole per ciascuna dimensione del tensore di input.
  • padding: Il tipo di algoritmo di padding da utilizzare.

Attributi opzionali (vedere Attrs ):

  • include_batch_in_index: Indica se includere la dimensione batch nell'indice appiattito di argmax .

Resi:

  • Output output: Il tensore di output massimo aggregato.
  • Output argmax: 4-D. Gli indici appiattiti dei valori massimi scelti per ciascun output.

Costruttori e distruttori

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)

attributi pubblici

argmax
operation
output

Funzioni statiche pubbliche

IncludeBatchInIndex (bool x)
Targmax (DataType x)

Strutture

tensorflow::ops::MaxPoolWithArgmax::Attrs

Impostatori di attributi opzionali per MaxPoolWithArgmax .

attributi pubblici

argmax

::tensorflow::Output argmax

operazione

Operation operation

produzione

::tensorflow::Output output

Funzioni pubbliche

MaxPoolWithArgmax

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

MaxPoolWithArgmax

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

Funzioni statiche pubbliche

IncludeBatchInIndex

Attrs IncludeBatchInIndex(
  bool x
)

Targmax

Attrs Targmax(
  DataType x
)