structure DML_ACTIVATION_SOFTMAX_OPERATOR_DESC (directml.h)
Exécute une fonction d’activation softmax sur InputTensor, en plaçant le résultat dans l’élément correspondant de OutputTensor.
For 1-D InputTensor:
// Let x[i] be the current element in the InputTensor, and j be the total number of elements in the InputTensor
f(x[i]) = exp(x[i]) / sum(exp(x[0]), ..., exp(x[j-1]))
Où exp(x) est la fonction d’exponentiation naturelle.
Syntaxe
struct DML_ACTIVATION_SOFTMAX_OPERATOR_DESC {
const DML_TENSOR_DESC *InputTensor;
const DML_TENSOR_DESC *OutputTensor;
};
Membres
InputTensor
Type : const DML_TENSOR_DESC*
Tenseur d’entrée à partir duquel lire. 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.
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 |