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
.
Resi:
- 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 )