Restez organisé à l'aide des collections Enregistrez et classez les contenus selon vos préférences.

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