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	:: tensorflow::Output
operation	Operation
output	:: tensorflow::Output

Fonctions statiques publiques
IncludeBatchInIndex (bool x)	Attrs
Targmax (DataType x)	Attrs

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
)