tensorflow::ops::MaxPoolWithArgmax

Effectue un regroupement maximal sur l'entrée et renvoie à 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] devient 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 compris dans [0, height) x [0, width) avant aplatissement, même en présence de marge intérieure et si la valeur mathématiquement correcte est en dehors de cet intervalle (négative ou trop grande). Il s'agit d'un bug, mais sa correction est complexe et difficilement compatible avec les versions précédentes, notamment à cause de l'aplatissement.

Arguments :

• portée : un objet Scope
• Entrée : 4D avec la forme [batch, height, width, channels] . Entrée à regrouper.
• ksize : La taille de la fenêtre pour chaque dimension du tenseur d’entrée.
• pas : Le pas de la fenêtre glissante pour chaque dimension du tenseur d'entrée.
• padding : le type d’algorithme de padding à utiliser.

Attributs optionnels (voir Attrs ) :

• include_batch_in_index : Indique s’il faut inclure la dimension batch dans l’index aplati de argmax .

Retours :

• Output : Le tenseur de sortie max poolé.
• Output argmax : 4D. 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
)