tensorflow :: operaciones :: MaxPoolWithArgmax

#include <nn_ops.h>

Realiza un agrupamiento máximo en la entrada y genera valores máximos e índices.

Resumen

Los índices en argmax se aplanan, de modo que un valor máximo en la posición [b, y, x, c] convierte en un índice aplanado: (y * width + x) * channels + c si include_batch_in_index es False; ((b * height + y) * width + x) * channels + c si include_batch_in_index es True.

Los índices devueltos siempre están en [0, height) x [0, width) antes del aplanamiento, incluso si se trata de un relleno y la respuesta matemáticamente correcta está afuera (ya sea negativa o demasiado grande). Esto es un error, pero solucionarlo es difícil de hacer de una manera segura compatible con versiones anteriores, especialmente debido al aplanamiento.

Argumentos:

  • alcance: un objeto de alcance
  • entrada: 4-D con forma [batch, height, width, channels] . Entrada para agrupar.
  • ksize: el tamaño de la ventana para cada dimensión del tensor de entrada.
  • strides: El paso de la ventana deslizante para cada dimensión del tensor de entrada.
  • padding: el tipo de algoritmo de relleno que se utilizará.

Atributos opcionales (consulte Attrs ):

  • include_batch_in_index: si se debe incluir la dimensión del lote en el índice plano de argmax .

Devoluciones:

  • Output salida: el tensor de salida combinado máximo.
  • Output argmax: 4-D. Los índices aplanados de los valores máximos elegidos para cada salida.

Constructores y Destructores

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

Funciones estáticas públicas

IncludeBatchInIndex (bool x)
Targmax (DataType x)

Estructuras

tensorflow :: ops :: MaxPoolWithArgmax :: Attrs

Establecedores de atributos opcionales para MaxPoolWithArgmax .

Atributos públicos

argmax

::tensorflow::Output argmax

operación

Operation operation

producción

::tensorflow::Output output

Funciones publicas

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
)

Funciones estáticas públicas

IncluirBatchInIndex

Attrs IncludeBatchInIndex(
  bool x
)

Targmax

Attrs Targmax(
  DataType x
)