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 )