flux tensoriel : : opérations : : MaxPoolAvecArgmax

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 .

Retour:

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

Attributs publics

::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
)