fluxo tensor:: ops:: MaxPoolWithArgmax
#include <nn_ops.h>
Executa o pooling máximo na entrada e gera valores máximos e índices.
Resumo
Os índices em argmax
são nivelados, de modo que um valor máximo na posição [b, y, x, c]
se torna um índice nivelado: (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 nivelamento, 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 segura e compatível com versões anteriores, especialmente devido ao nivelamento.
Argumentos:
- escopo: um objeto Escopo
- entrada: 4-D com forma
[batch, height, width, channels]
. Entrada para agrupar. - ksize: O tamanho da janela para cada dimensão do tensor de entrada.
- passos: O passo da janela deslizante para cada dimensão do tensor de entrada.
- preenchimento: o tipo de algoritmo de preenchimento a ser usado.
Atributos opcionais (veja Attrs
):
- include_batch_in_index: se deve incluir a dimensão do lote no índice nivelado de
argmax
.
Retorna:
- Saída
Output
: o tensor de saída máximo agrupado. - Argmax
Output
: 4-D. Os índices nivelados 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) |
Estruturas | |
---|---|
tensorflow:: ops:: MaxPoolWithArgmax:: Attrs | Configuradores de atributos opcionais para MaxPoolWithArgmax . |
Atributos públicos
argmax
::tensorflow::Output argmax
operação
Operation operation
saída
::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
IncluirBatchInIndex
Attrs IncludeBatchInIndex( bool x )
Targmax
Attrs Targmax( DataType x )