tensorflow::ops::FusedPadConv2D

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 par remplissage spatial est fusionnée avec la recherche im2col, mais sans le filtrage bilinéaire requis pour le redimensionnement. La fusion du remplissage évite d'écrire les résultats intermédiaires sous forme de tenseurs entiers, réduisant ainsi la consommation de mémoire. De plus, la fusion des calculs de transformation permet de gagner en latence. L'attribut data_format de Conv2D n'est pas pris en charge par cette opération ; l'ordre « NHWC » est utilisé à la place. En interne, cette opération utilise un seul tampon temporaire par graphe, ce qui signifie qu'elle sera bloquante 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] .
• 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.

Retours :

• 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