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

DML_FEATURE_LEVEL_2_0 et versions ultérieures

DML_FEATURE_LEVEL_1_0 et versions ultérieures

Configuration requise

Voir aussi

DML_ACTIVATION_LOG_SOFTMAX_OPERATOR_DESC structure

DML_ARGMAX_OPERATOR_DESC structure