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 False; ((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. -
Output
argmax: 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 )