tensorflow :: opérations :: MaxPoolWithArgmax

#include <nn_ops.h>

Effectue une mise en commun maximale sur l'entrée et génère à la fois des valeurs maximales et des indices.

Résumé

Les indices dans argmax sont aplatis, de sorte qu'une valeur maximale à la position [b, y, x, c] devient 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 vaut True.

Les indices renvoyés sont toujours entre [0, height) x [0, width) avant l'aplatissement, même si le 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 bogue, mais il est difficile de le corriger de manière sûre et compatible avec les versions antérieures, en particulier en raison de l'aplatissement.

Arguments:

  • scope: un objet Scope
  • entrée: 4-D avec forme [batch, height, width, channels] . Entrée pour regrouper.
  • 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:

  • Output sortie: Le tenseur de sortie maximum mis en commun.
  • Output argmax: 4-D. Les indices aplatis des valeurs max 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)

Structs

tensorflow :: ops :: MaxPoolWithArgmax :: Attrs

Paramètres d' attribut facultatifs pour MaxPoolWithArgmax .

Attributs publics

argmax

::tensorflow::Output argmax

opération

Operation operation

production

::tensorflow::Output output

Fonctions publiques

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
)

Fonctions statiques publiques

IncludeBatchInIndex

Attrs IncludeBatchInIndex(
  bool x
)

Targmax

Attrs Targmax(
  DataType x
)