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 )