Partager via

Structure DML_ACTIVATION_SOFTMAX1_OPERATOR_DESC (directml.h)

  • Article

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

Voir aussi