flux tensoriel : : opérations : : MaxPoolAvecArgmax

#include <nn_ops.h>

Effectue un pooling maximum sur l'entrée et génère à la fois les valeurs maximales et les indices.

Résumé

Les indices dans argmax sont aplatis, de sorte qu'une valeur maximale à la position [b, y, x, c] devienne un index aplati : (y * width + x) * channels + c si include_batch_in_index est False ; ((b * height + y) * width + x) * channels + c si include_batch_in_index est True.

Les indices renvoyés sont toujours dans [0, height) x [0, width) avant aplatissement, même si un remplissage est impliqué et que la réponse mathématiquement correcte est à l'extérieur (soit négative, soit trop grande). Il s'agit d'un bug, mais il est difficile de le corriger de manière sécurisée et rétrocompatible, notamment en raison de l'aplatissement.

Arguments :

  • scope : un objet Scope
  • entrée : 4-D avec forme [batch, height, width, channels] . Entrée pour mettre en commun.
  • ksize : La taille de la fenêtre pour chaque dimension du tenseur d'entrée.
  • foulées : la foulée de la fenêtre glissante pour chaque dimension du tenseur d'entrée.
  • padding : Le type d’algorithme de remplissage à utiliser.

Attributs facultatifs (voir Attrs ) :

  • include_batch_in_index : s'il faut inclure la dimension du lot dans l'index aplati de argmax .

Retours :

  • Sortie Output : le tenseur de sortie regroupé maximum.
  • Argmax Output : 4-D. Les indices aplatis des valeurs maximales choisies pour chaque sortie.

Constructeurs et Destructeurs

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)

Attributs publics

argmax
operation
output

Fonctions statiques publiques

IncludeBatchInIndex (bool x)
Targmax (DataType x)

Structures

tensorflow :: ops :: MaxPoolWithArgmax :: Attrs

Setters d'attributs facultatifs pour MaxPoolWithArgmax .

Attributs publics

argmax 

::tensorflow::Output argmax

opération 

Operation operation

sortir 

::tensorflow::Output output

Fonctions publiques

MaxPoolAvecArgmax 

 MaxPoolWithArgmax(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  const gtl::ArraySlice< int > & ksize,
  const gtl::ArraySlice< int > & strides,
  StringPiece padding
)

MaxPoolAvecArgmax 

 MaxPoolWithArgmax(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  const gtl::ArraySlice< int > & ksize,
  const gtl::ArraySlice< int > & strides,
  StringPiece padding,
  const MaxPoolWithArgmax::Attrs & attrs
)

Fonctions statiques publiques

InclureBatchInIndex 

Attrs IncludeBatchInIndex(
  bool x
)

Targmax 

Attrs Targmax(
  DataType x
)