tensorflow :: opérations :: MaxPoolWithArgmax

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.

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
)