tensorflow :: opérations :: FusedPadConv2D
#include <nn_ops.h>
Effectue un remplissage en tant que prétraitement lors d'une convolution.
Résumé
Similaire à FusedResizeAndPadConv2d, cette opération permet une implémentation optimisée où l'étape de transformation de remplissage spatial est fusionnée avec la recherche im2col, mais dans ce cas sans le filtrage bilinéaire requis pour le redimensionnement. La fusion du remplissage évite d'avoir à écrire les résultats intermédiaires sous forme de tenseurs entiers, réduisant 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 l'ordre 'NHWC' est utilisé à la place. En interne, cette opération utilise un seul tampon de travail par graphe, 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]
. - 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.
Retour:
-
Output
: le tenseur de sortie.
Constructeurs et destructeurs | |
---|---|
FusedPadConv2D (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input paddings, :: tensorflow::Input filter, StringPiece mode, const gtl::ArraySlice< int > & strides, StringPiece padding) |
Attributs publics | |
---|---|
operation | |
output |
Fonctions publiques | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
Attributs publics
opération
Operation operation
production
::tensorflow::Output output
Fonctions publiques
FusedPadConv2D
FusedPadConv2D( const ::tensorflow::Scope & scope, ::tensorflow::Input input, ::tensorflow::Input paddings, ::tensorflow::Input filter, StringPiece mode, const gtl::ArraySlice< int > & strides, StringPiece padding )
nœud
::tensorflow::Node * node() const
operator :: tensorflow :: Input
operator::tensorflow::Input() const
operator :: tensorflow :: Output
operator::tensorflow::Output() const