DML_ACTIVATION_HARDMAX_OPERATOR_DESC structure (directml.h)
Exécute une fonction hardmax sur chaque élément de InputTensor, en plaçant le résultat dans l’élément correspondant de OutputTensor.
L’opérateur calcule la valeur hardmax (1 pour la première occurrence de la valeur la plus élevée dans la couche et 0 pour toutes les autres valeurs) de chaque ligne dans l’entrée donnée.
Syntaxe
struct DML_ACTIVATION_HARDMAX_OPERATOR_DESC {
const DML_TENSOR_DESC *InputTensor;
const DML_TENSOR_DESC *OutputTensor;
};
Membres
InputTensor
Type : const DML_TENSOR_DESC*
Tenseur à lire pour l’entrée. Ce tenseur ne doit pas avoir un rang effectif supérieur à 2. Le rang effectif d’un tenseur est dimensionCount du tenseur, à l’exclusion des dimensions les plus à gauche de la taille 1. Par exemple, une taille de tenseur de { 1, 1, BatchCount, Width } est valide et équivaut à un tenseur de tailles { BatchCount, Width }.
OutputTensor
Type : const DML_TENSOR_DESC*
Tenseur de sortie dans lequel écrire les résultats.
Remarques
L’opérateur calcule les valeurs hardmax (1 pour la première valeur maximale et 0 pour toutes les autres) pour chaque couche du lot de l’entrée donnée. L’entrée est un tenseur 2D (Tensor) de taille (batch_size x input_feature_dimensions). Le tenseur de sortie a la même forme et contient les valeurs hardmax de l’entrée correspondante.
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 des tenseurs
DML_FEATURE_LEVEL_3_0 et versions ultérieures
| Tenseur | Genre | Nombre de dimensions pris en charge | Types de données pris en charge |
|---|---|---|---|
| InputTensor | Entrée | 1 à 8 | FLOAT32, FLOAT16 |
| OutputTensor | Sortie | 1 à 8 | FLOAT32, FLOAT16 |
DML_FEATURE_LEVEL_2_0 et versions ultérieures
| Tenseur | Genre | Nombre de dimensions pris en charge | Types de données pris en charge |
|---|---|---|---|
| InputTensor | Entrée | 4 à 5 | FLOAT32, FLOAT16 |
| OutputTensor | Sortie | 4 à 5 | FLOAT32, FLOAT16 |
DML_FEATURE_LEVEL_1_0 et versions ultérieures
| Tenseur | Genre | Nombre de dimensions pris en charge | Types de données pris en charge |
|---|---|---|---|
| InputTensor | Entrée | 4 | FLOAT32, FLOAT16 |
| OutputTensor | Sortie | 4 | FLOAT32, FLOAT16 |
Configuration requise
| Condition requise | Valeur |
|---|---|
| En-tête | directml.h |