Notes
L’accès à cette page nécessite une autorisation. Vous pouvez essayer de vous connecter ou de modifier des répertoires.
L’accès à cette page nécessite une autorisation. Vous pouvez essayer de modifier des répertoires.
Exécute une fonction d’activation d’unité linéaire rectifiée (ReLU) seuil sur chaque élément dans InputTensor, en plaçant le résultat dans l’élément correspondant de OutputTensor.
f(x) = x, if x > Alpha
0, otherwise
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_THRESHOLDED_RELU_OPERATOR_DESC {
const DML_TENSOR_DESC *InputTensor;
const DML_TENSOR_DESC *OutputTensor;
FLOAT Alpha;
};
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
Seuil de la fonction. Une valeur par défaut classique pour cette valeur est 1.0.
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 |