structure DML_ACTIVATION_SCALED_ELU_OPERATOR_DESC (directml.h)

Article
08/23/2023

Exécute une fonction d’activation d’unité linéaire exponentielle (ELU) mise à l’échelle sur chaque élément dans InputTensor, en plaçant le résultat dans l’élément correspondant de OutputTensor.

f(x) = Gamma * x,                        if x > 0
       Gamma * (Alpha * exp(x) - Alpha), otherwise

Où exp(x) est la fonction d’exponentiation naturelle.

Cet opérateur prend en charge l’exécution sur place, ce qui signifie que le tenseur de sortie est autorisé à alias InputTensor pendant la liaison.

Syntaxe

struct DML_ACTIVATION_SCALED_ELU_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
  FLOAT                 Alpha;
  FLOAT                 Gamma;
};

Membres

InputTensor

Type : const DML_TENSOR_DESC*

Tenseur d’entrée à partir duquel lire.

OutputTensor

Type : const DML_TENSOR_DESC*

Tenseur de sortie dans lequel écrire les résultats.

Alpha

Type : FLOAT

Valeur alpha. Une valeur par défaut classique pour cette valeur est 1,6732.

Gamma

Type : FLOAT

Valeur de gamma. Une valeur par défaut classique pour cette valeur est 1,0507.

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_0 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_2_0 et versions ultérieures

Tenseur	Type	Nombre de dimensions pris en charge	Types de données pris en charge
InputTensor	Entrée	4 à 5	FLOAT32, FLOAT16
OutputTensor	Output	4 à 5	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

Voir aussi

DML_ACTIVATION_ELU_OPERATOR_DESC

Partager via