tensorflow::ops::Conv2D::Attrs

#include <nn_ops.h>

Définiteurs d'attributs optionnels pour Conv2D .

Résumé

attributs publics

data_format_ = "NHWC"
StringPiece
dilations_ = Default_dilations()
gtl::ArraySlice< int >
explicit_paddings_ = {}
gtl::ArraySlice< int >
use_cudnn_on_gpu_ = true
bool

fonctions publiques

DataFormat (StringPiece x)
TF_MUST_USE_RESULT Attrs
Spécifiez le format des données d'entrée et de sortie.
Dilations (const gtl::ArraySlice< int > & x)
TF_MUST_USE_RESULT Attrs
Tenseur 1D de longueur 4.
ExplicitPaddings (const gtl::ArraySlice< int > & x)
TF_MUST_USE_RESULT Attrs
Si padding est "EXPLICIT" , la liste des montants de remplissage explicites.
UseCudnnOnGpu (bool x)
TF_MUST_USE_RESULT Attrs
Par défaut, la valeur est « vrai ».

attributs publics

format_de_données

StringPiece tensorflow::ops::Conv2D::Attrs::data_format_ = "NHWC"

dilatations

gtl::ArraySlice< int > tensorflow::ops::Conv2D::Attrs::dilations_ = Default_dilations()

rembourrages_ explicites

gtl::ArraySlice< int > tensorflow::ops::Conv2D::Attrs::explicit_paddings_ = {}

utiliser_cudnn_sur_gpu_

bool tensorflow::ops::Conv2D::Attrs::use_cudnn_on_gpu_ = true

fonctions publiques

Format de données

TF_MUST_USE_RESULT Attrs tensorflow::ops::Conv2D::Attrs::DataFormat(
  StringPiece x
)

Spécifiez le format des données d'entrée et de sortie.

Avec le format par défaut « NHWC », les données sont stockées dans l’ordre suivant : [lot, hauteur, largeur, canaux]. Le format « NCHW » permet quant à lui de stocker les données dans l’ordre suivant : [lot, canaux, hauteur, largeur].

Par défaut, la valeur est « NHWC »

Dilatations

TF_MUST_USE_RESULT Attrs tensorflow::ops::Conv2D::Attrs::Dilations(
  const gtl::ArraySlice< int > & x
)

Tenseur 1D de longueur 4.

Le facteur de dilatation pour chaque dimension de input . Si k > 1, k-1 cellules seront ignorées entre chaque élément de filtre sur cette dimension. L'ordre des dimensions est déterminé par la valeur de data_format (voir ci-dessus pour plus de détails). Les dilatations des dimensions batch et depth doivent être égales à 1.

Valeur par défaut : [1, 1, 1, 1]

Rembourrages explicites

TF_MUST_USE_RESULT Attrs tensorflow::ops::Conv2D::Attrs::ExplicitPaddings(
  const gtl::ArraySlice< int > & x
)

Si padding est "EXPLICIT" , la liste des montants de remplissage explicites.

Pour la i-ème dimension, la quantité de remplissage insérée avant et après la dimension est respectivement explicit_paddings[2 * i] et explicit_paddings[2 * i + 1] . Si padding n'est pas "EXPLICIT" , explicit_paddings doit être vide.

Valeur par défaut : []

Utiliser Cudnn sur GPU

TF_MUST_USE_RESULT Attrs tensorflow::ops::Conv2D::Attrs::UseCudnnOnGpu(
  bool x
)

Par défaut, la valeur est « vrai ».