flux tensoriel : : opérations : : Conv2D : Attributs
#include <nn_ops.h>
Setters d'attributs facultatifs 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 de données des données d'entrée et de sortie. |
Dilations (const gtl::ArraySlice< int > & x) | TF_MUST_USE_RESULT Attrs Tenseur 1-D 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 La valeur par défaut est vrai. |
Attributs publics
format_données_
StringPiece tensorflow::ops::Conv2D::Attrs::data_format_ = "NHWC"
dilatations_
gtl::ArraySlice< int > tensorflow::ops::Conv2D::Attrs::dilations_ = Default_dilations()
explicit_paddings_
gtl::ArraySlice< int > tensorflow::ops::Conv2D::Attrs::explicit_paddings_ = {}
use_cudnn_on_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 de données 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 : [lot, hauteur, largeur, canaux]. Alternativement, le format pourrait être « NCHW », l'ordre de stockage des données étant : [lot, canaux, hauteur, largeur].
La valeur par défaut est "NHWC"
Dilatations
TF_MUST_USE_RESULT Attrs tensorflow::ops::Conv2D::Attrs::Dilations( const gtl::ArraySlice< int > & x )
Tenseur 1-D de longueur 4.
Le facteur de dilatation pour chaque dimension d' input
. Si défini sur k > 1, il y aura k-1 cellules 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 dans les dimensions du lot et de la profondeur doivent être de 1.
La valeur par défaut est [1, 1, 1, 1]
Remplissages 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.
La valeur par défaut est []
UtiliserCudnnOnGpu
TF_MUST_USE_RESULT Attrs tensorflow::ops::Conv2D::Attrs::UseCudnnOnGpu( bool x )
La valeur par défaut est vrai.