flux tensoriel : : opérations : : FusedPadConv2D

Effectue un remplissage comme prétraitement lors d'une convolution.

Résumé

Semblable à FusedResizeAndPadConv2d, cette opération permet une implémentation optimisée dans laquelle l'étape de transformation du 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 ainsi la pression sur la 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 graphique, ce qui signifie qu'elle se bloquera si plusieurs versions sont exécutées en parallèle. En effet, cet opérateur est avant tout une optimisation visant à 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 : une matrice à deux colonnes spécifiant les tailles de remplissage. Le nombre de lignes doit être le même que le rang de 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 glissante 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.

Attributs publics

Operation operation

::tensorflow::Output output

 FusedPadConv2D(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input paddings,
  ::tensorflow::Input filter,
  StringPiece mode,
  const gtl::ArraySlice< int > & strides,
  StringPiece padding
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const