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
)