tensorflow::ops::MaxPoolWithArgmax
#include <nn_ops.h>Executa o agrupamento máximo (max pooling) na entrada e retorna tanto os valores máximos quanto os índices.
Resumo
Os índices em argmax são achatados, de modo que um valor máximo na posição [b, y, x, c] se torna o í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 que haja preenchimento e a resposta matematicamente correta esteja fora desse intervalo (seja negativa ou muito grande). Isso é um bug, mas corrigi-lo de forma segura e retrocompatível é difícil, especialmente devido ao achatamento.
Argumentos:
- escopo: Um objeto de escopo
- Entrada: 4-D com formato
[batch, height, width, channels]. Entrada para agrupamento. - ksize: O tamanho da janela para cada dimensão do tensor de entrada.
- strides: 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 (ver Attrs ):
- include_batch_in_index: Indica se a dimensão do lote deve ser incluída no índice achatado de
argmax.
Devoluções:
-
Output: O tensor de saída com agrupamento máximo. -
Outputargmax: 4-D. Os índices achatados dos valores máximos escolhidos para cada saída.
Construtores e Destrutores | |
|---|---|
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 | Definidores 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
IncluirLoteNoÍndice
Attrs IncludeBatchInIndex( bool x )
Targmax
Attrs Targmax( DataType x )