tensorflow :: operaciones :: MaxPoolWithArgmax

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.

Atributos públicos

::tensorflow::Output argmax

Operation operation

::tensorflow::Output output

 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
)

Attrs IncludeBatchInIndex(
  bool x
)

Attrs Targmax(
  DataType x
)