tensorflow::ops::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 d'effectuer des transformations spatiales plus efficacement lors de l'étape d'empaquetage d'une convolution. Cette opération permet donc une implémentation optimisée où ces étapes sont fusionnées. Cela évite d'écrire les résultats intermédiaires sous forme de tenseurs complets, réduisant ainsi la consommation de mémoire, et permet de gagner en latence grâce à la fusion des calculs de transformation. L'attribut `data_format` de `Conv2D` n'est pas pris en charge par cette opération et utilise par défaut l'ordre `NHWC`. En interne, cette opération utilise un seul tampon temporaire par graphe, ce qui signifie qu'elle sera bloquée si plusieurs instances sont exécutées en parallèle. En effet, cet opérateur vise principalement à minimiser l'utilisation de la mémoire.

Arguments :

  • portée : un objet Scope
  • entrée : 4-D avec forme [batch, in_height, in_width, in_channels] .
  • taille : Un tenseur 1D int32 à 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 identique au rang de input .
  • filtre : 4D avec forme [filter_height, filter_width, in_channels, out_channels] .
  • strides : 1-D de longueur 4. Le pas de la fenêtre glissante pour chaque dimension de input . Doit être dans le même ordre que la dimension spécifiée avec le format.
  • padding : le type d’algorithme de padding à utiliser.

Attributs optionnels (voir Attrs ) :

  • resize_align_corners : Si la valeur est « true », les centres des quatre pixels d’angle des tenseurs d’entrée et de sortie sont alignés, préservant ainsi les valeurs de ces pixels. La valeur par défaut est « false ».

Retours :

  • 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)

Structures

tensorflow::ops::FusedResizeAndPadConv2D::Attrs

Définiteurs d'attributs optionnels pour FusedResizeAndPadConv2D .

attributs publics

opération

Operation operation

sortir

::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 

opérateur::tensorflow::Entrée

 operator::tensorflow::Input() const 

opérateur::tensorflow::Sortie

 operator::tensorflow::Output() const 

fonctions statiques publiques

RedimensionnerAlignerLesCoins

Attrs ResizeAlignCorners(
  bool x
)