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 )