Structure DML_ACTIVATION_SOFTMAX1_OPERATOR_DESC (directml.h)
Exécute une fonction d’activation softmax sur InputTensor, en plaçant le résultat dans l’élément OutputTensor correspondant.
For 1-D InputTensor:
// Let x[i] to 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 naturelle d’élévation à une puissance.
Important
Cette API est disponible dans le cadre du package redistribuable autonome DirectML (voir Microsoft.AI.DirectML version 1.9 et ultérieures). Consultez également l’historique des versions DirectML.
Syntaxe
struct DML_ACTIVATION_SOFTMAX1_OPERATOR_DESC
{
const DML_TENSOR_DESC* InputTensor;
const DML_TENSOR_DESC* OutputTensor;
UINT AxisCount;
_Field_size_(AxisCount) const UINT* Axes;
};
Membres
InputTensor
Type : const DML_TENSOR_DESC*
Tenseur d’entrée à partir duquel effectuer la lecture.
OutputTensor
Type : const DML_TENSOR_DESC*
Tenseur de sortie dans lequel le résultat doit être écrit.
AxisCount
Type : UINT
Nombre d’axes pour calculer la fonction reduce sum. Ce champ détermine la taille du tableau Axes.
Axes
Type : _Field_size_(AxisCount) const UINT*
Axes le long desquels réduire la somme. Les valeurs doivent être comprise dans la plage [0, InputTensor.DimensionCount - 1].
Notes
Cet opérateur équivaut à DML_ACTIVATION_SOFTMAX_OPERATOR_DESC lorsque AxisCount == 1 et Axes == {DimensionCount - 1}.
Disponibilité
Cet opérateur a été introduit dans DML_FEATURE_LEVEL_5_1.
Contraintes de tenseur
InputTensor et OutputTensor doivent avoir les mêmes DataType, DimensionCount et Sizes.
Prise en charge des tenseurs
| Tenseur | Genre | Nombre de dimensions pris en charge | Types de données pris en charge |
|---|---|---|---|
| InputTensor | Input | 1 à 8 | FLOAT32, FLOAT16 |
| OutputTensor | Sortie | 1 à 8 | FLOAT32, FLOAT16 |
Spécifications
| En-tête | directml.h |