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:
-
Outputoutput: Il tensore di output massimo aggregato. -
Outputargmax: 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 )