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 )