structure DML_LP_NORMALIZATION_OPERATOR_DESC (directml.h)
Effectue une fonction de normalisation Lp le long de l’axe spécifié du tenseur d’entrée.
Syntaxe
struct DML_LP_NORMALIZATION_OPERATOR_DESC {
const DML_TENSOR_DESC *InputTensor;
const DML_TENSOR_DESC *OutputTensor;
UINT Axis;
FLOAT Epsilon;
UINT P;
};
Membres
InputTensor
Type : const DML_TENSOR_DESC*
Tenseur contenant les données d’entrée.
OutputTensor
Type : const DML_TENSOR_DESC*
Tenseur dans lequel écrire les résultats. Les tailles de ce tenseur doivent correspondre au inputTensor.
Axis
Type : UINT
Axe sur lequel appliquer la normalisation.
Epsilon
Type : FLOAT
Valeur epsilon à utiliser pour éviter la division par zéro. La valeur par défaut 0,00001 est recommandée.
P
Type : UINT
Ordre de la normalisation (1 ou 2).
Disponibilité
Cet opérateur a été introduit dans DML_FEATURE_LEVEL_1_0
.
Contraintes tensoriels
InputTensor et OutputTensor doivent avoir les mêmes DataType, DimensionCount et Sizes.
Prise en charge de Tensor
DML_FEATURE_LEVEL_3_1 et versions ultérieures
Tenseur | Type | Nombre de dimensions pris en charge | Types de données pris en charge |
---|---|---|---|
InputTensor | Entrée | 1 à 8 | FLOAT32, FLOAT16 |
OutputTensor | Output | 1 à 8 | FLOAT32, FLOAT16 |
DML_FEATURE_LEVEL_1_0 et versions ultérieures
Tenseur | Type | Nombre de dimensions pris en charge | Types de données pris en charge |
---|---|---|---|
InputTensor | Entrée | 4 | FLOAT32, FLOAT16 |
OutputTensor | Output | 4 | FLOAT32, FLOAT16 |
Spécifications
En-tête | directml.h |