tensorflow :: opérations :: FusedResizeAndPadConv2D

#include <nn_ops.h>

Effectue un redimensionnement et un remplissage en tant que prétraitement lors d'une convolution.

Résumé

Il est souvent possible de faire 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, ce qui réduit la pression 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 prend 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 principalement une optimisation pour 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 Tensor 1-D int32 de 2 éléments: new_height, new_width . La nouvelle taille des images.
  • paddings: matrice à deux colonnes spécifiant les tailles de remplissage. Le nombre de lignes doit être le même que le rang de l' 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 coulissante 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 true, les centres des 4 pixels d'angle des tenseurs d'entrée et de sortie sont alignés, en préservant les valeurs aux pixels d'angle. La valeur par défaut est false.

Retour:

  • Output : le tenseur de sortie.

Constructeurs et destructeurs

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)

Attributs publics

operation
output

Fonctions publiques

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

Fonctions statiques publiques

ResizeAlignCorners (bool x)

Structs

tensorflow :: ops :: FusedResizeAndPadConv2D :: Attrs

Définisseurs d' attributs facultatifs pour FusedResizeAndPadConv2D .

Attributs publics

opération

Operation operation

production

::tensorflow::Output output

Fonctions publiques

FusedResizeAndPadConv2D

 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

 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
)

nœud

::tensorflow::Node * node() const 

operator :: tensorflow :: Input

 operator::tensorflow::Input() const 

operator :: tensorflow :: Output

 operator::tensorflow::Output() const 

Fonctions statiques publiques

RedimensionnerAlignerCoins

Attrs ResizeAlignCorners(
  bool x
)