tensorflow :: ops :: MaxPoolWithArgmax
#include <nn_ops.h>
Executa o pooling máximo na entrada e emite tanto os valores máximos quanto os índices.
Resumo
Os índices em argmax
são achatados, de forma que um valor máximo na posição [b, y, x, c]
se torna índice achatado: (y * width + x) * channels + c
se include_batch_in_index
for False; ((b * height + y) * width + x) * channels + c
se include_batch_in_index
for True.
Os índices retornados estão sempre em [0, height) x [0, width)
antes do achatamento, mesmo se o preenchimento estiver envolvido e a resposta matematicamente correta estiver fora (negativa ou muito grande). Este é um bug, mas é difícil corrigi-lo de uma forma compatível com versões anteriores, especialmente devido ao achatamento.
Argumentos:
- escopo: um objeto Scope
- entrada: 4-D com forma
[batch, height, width, channels]
. Entrada para finalizar a piscina. - ksize: O tamanho da janela para cada dimensão do tensor de entrada.
- strides: O stride da janela deslizante para cada dimensão do tensor de entrada.
- preenchimento: o tipo de algoritmo de preenchimento a ser usado.
Atributos opcionais (consulte Attrs
):
- include_batch_in_index: se a dimensão do lote deve ser incluída no índice achatado de
argmax
.
Retorna:
-
Output
saída: O tensor de saída máxima em pool. -
Output
argmax: 4-D. Os índices achatados dos valores máximos escolhidos para cada saída.
Construtores e Destruidores | |
---|---|
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 |
Funções estáticas públicas | |
---|---|
IncludeBatchInIndex (bool x) | |
Targmax (DataType x) |
Structs | |
---|---|
tensorflow :: ops :: MaxPoolWithArgmax :: Attrs | Configuradores de atributos opcionais para MaxPoolWithArgmax . |
Atributos públicos
argmax
::tensorflow::Output argmax
Operação
Operation operation
resultado
::tensorflow::Output output
Funções públicas
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 )
Funções estáticas públicas
IncludeBatchInIndex
Attrs IncludeBatchInIndex( bool x )
Targmax
Attrs Targmax( DataType x )