flujo tensor:: operaciones:: MaxPoolConArgmax
#include <nn_ops.h>Realiza una agrupación máxima en la entrada y genera valores e índices máximos.
Resumen
Los índices en argmax están aplanados, de modo que un valor máximo en la posición [b, y, x, c] se convierte en índice aplanado: (y * width + x) * channels + c si include_batch_in_index es Falso; ((b * height + y) * width + x) * channels + c si include_batch_in_index es Verdadero.
Los índices devueltos siempre están en [0, height) x [0, width) antes del aplanamiento, incluso si se trata de relleno y la respuesta matemáticamente correcta está afuera (ya sea negativa o demasiado grande). Esto es un error, pero es difícil solucionarlo de una manera segura y compatible con versiones anteriores, especialmente debido al aplanamiento.
Argumentos:
- alcance: un objeto de alcance
- entrada: 4-D con forma
[batch, height, width, channels]. Entrada para agrupar. - ksize: el tamaño de la ventana para cada dimensión del tensor de entrada.
- zancadas: la zancada de la ventana deslizante para cada dimensión del tensor de entrada.
- padding: el tipo de algoritmo de relleno que se utilizará.
Atributos opcionales (ver Attrs ):
- include_batch_in_index: si se debe incluir la dimensión del lote en el índice aplanado de
argmax.
Devoluciones:
- Salida
Output: el tensor de salida agrupado máximo. -
Outputargmax: 4-D. Los índices aplanados de los valores máximos elegidos para cada salida.
Constructores y destructores | |
|---|---|
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) |
Atributos públicos | |
|---|---|
argmax | |
operation | |
output | |
Funciones estáticas públicas | |
|---|---|
IncludeBatchInIndex (bool x) | |
Targmax (DataType x) | |
estructuras | |
|---|---|
| tensorflow:: operaciones:: MaxPoolWithArgmax:: Atributos | Configuradores de atributos opcionales para MaxPoolWithArgmax . |
Atributos públicos
argmax
::tensorflow::Output argmax
operación
Operation operation
producción
::tensorflow::Output output
Funciones públicas
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 )
Funciones estáticas públicas
IncluirBatchInIndex
Attrs IncludeBatchInIndex( bool x )
Targmax
Attrs Targmax( DataType x )