tensoreflusso:: ops:: MaxPoolConArgmax

#include <nn_ops.h>

Esegue il pooling massimo 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 un 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 in [0, height) x [0, width) prima dell'appiattimento, anche se è coinvolto il riempimento e la risposta matematicamente corretta è esterna (negativa o troppo grande). Si tratta di un bug, ma è difficile risolverlo in modo sicuro e compatibile con le versioni precedenti, soprattutto a causa dell'appiattimento.

Argomenti:

  • scope: un oggetto Scope
  • input: 4-D con forma [batch, height, width, channels] . Ingresso per raggruppare.
  • ksize: la dimensione della finestra per ciascuna dimensione del tensore di input.
  • passi: il passo della finestra scorrevole per ogni dimensione del tensore di input.
  • riempimento: il tipo di algoritmo di riempimento da utilizzare.

Attributi facoltativi (vedi Attrs ):

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

Ritorna:

  • Output Output : il massimo tensore di output in pool.
  • Argmax Output : 4-D. Gli indici appiattiti dei valori massimi scelti per ciascuna uscita.

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 pubbliche statiche

IncludeBatchInIndex (bool x)
Targmax (DataType x)

Strutture

tensorflow:: ops:: MaxPoolWithArgmax:: Attrs

Setter di attributi facoltativi per MaxPoolWithArgmax .

Attributi pubblici

argmax

::tensorflow::Output argmax

operazione

Operation operation

produzione

::tensorflow::Output output

Funzioni pubbliche

MaxPoolConArgmax

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

MaxPoolConArgmax

 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 pubbliche statiche

IncludeBatchInIndex

Attrs IncludeBatchInIndex(
  bool x
)

Targmax

Attrs Targmax(
  DataType x
)