flux tensoriel : : opérations : : FusedResizeAndPadConv2D

Effectue un redimensionnement et un remplissage en guise de prétraitement lors d'une convolution.

Résumé

Il est souvent possible d'effectuer des transformations spatiales plus efficacement dans le cadre de l'étape de compression d'une convolution, cette opération permet donc une implémentation optimisée où ces étapes sont fusionnées. Cela évite d'avoir à écrire les résultats intermédiaires sous forme de tenseurs entiers, réduisant ainsi la pression sur la mémoire, et nous pouvons obtenir des gains de latence en fusionnant les calculs de transformation. L'attribut data_format pour Conv2D n'est pas pris en charge par cette opération et est par défaut l'ordre « NHWC ». En interne, cette opération utilise un seul tampon de travail par graphique, ce qui signifie qu'elle se bloquera si plusieurs versions sont exécutées en parallèle. En effet, cet opérateur est avant tout une optimisation visant à minimiser l'utilisation de la mémoire.

Arguments :

• scope : un objet Scope
• entrée : 4-D avec forme [batch, in_height, in_width, in_channels] .
• size : Un Tenseur 1-D int32 de 2 éléments : new_height, new_width . La nouvelle taille des images.
• paddings : une matrice à deux colonnes spécifiant les tailles de remplissage. Le nombre de lignes doit être le même que le rang de input .
• filtre : 4-D avec forme [filter_height, filter_width, in_channels, out_channels] .
• foulées : 1-D de longueur 4. La foulée de la fenêtre glissante pour chaque dimension d' input . Doit être dans le même ordre que la dimension spécifiée avec le format.
• padding : Le type d’algorithme de remplissage à utiliser.

Attributs facultatifs (voir Attrs ) :

• resize_align_corners : Si vrai, les centres des 4 pixels de coin des tenseurs d'entrée et de sortie sont alignés, préservant les valeurs aux pixels de coin. La valeur par défaut est false.

Retours :

• Output : Le tenseur de sortie.

Attributs publics

Operation operation

::tensorflow::Output output

 FusedResizeAndPadConv2D(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input size,
  ::tensorflow::Input paddings,
  ::tensorflow::Input filter,
  StringPiece mode,
  const gtl::ArraySlice< int > & strides,
  StringPiece padding
)

 FusedResizeAndPadConv2D(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input size,
  ::tensorflow::Input paddings,
  ::tensorflow::Input filter,
  StringPiece mode,
  const gtl::ArraySlice< int > & strides,
  StringPiece padding,
  const FusedResizeAndPadConv2D::Attrs & attrs
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const 

Attrs ResizeAlignCorners(
  bool x
)