structure DML_ACTIVATION_SOFTMAX1_OPERATOR_DESC (directml.h)

Exécute une fonction d’activation softmax sur InputTensor, plaçant le résultat dans l’élément correspondant de OutputTensor.

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 d’exposant naturelle.

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*

Capteur d’entrée à partir duquel lire.

OutputTensor

Type : const DML_TENSOR_DESC*

Le tensoriel de sortie dans lequel écrire les résultats.

AxisCount

Type : UINT

Nombre d’axes pour calculer la somme de réduction. 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 se trouver dans la plage [0, InputTensor.DimensionCount - 1].

Remarques

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 Tensor

InputTensor et OutputTensor doivent avoir les mêmes DataType, DimensionCount et Sizes.

Prise en charge de Tensor

Spécifications

Voir aussi

• DML_ACTIVATION_LOG_SOFTMAX1_OPERATOR_DESC