DML_ACTIVATION_LOG_SOFTMAX_OPERATOR_DESC structure (directml.h)
Exécute une fonction d’activation naturelle log-of-softmax sur chaque élément de InputTensor, en plaçant le résultat dans l’élément correspondant de OutputTensor.
// Let x_i be the current value in the axis, and j be the total number of elements along that axis.
f(x_i) = ln(exp(x_i) / sum(exp(x_0), ..., exp(x_j)))
Où exp(x) est la fonction d’exponentiation naturelle, et ln(x) est le logarithme naturel.
Syntaxe
struct DML_ACTIVATION_LOG_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 | Type | Nombre de dimensions pris en charge | Types de données pris en charge |
|---|---|---|---|
| InputTensor | Entrée | 1 à 8 | FLOAT32, FLOAT16 |
| OutputTensor | Output | 1 à 8 | FLOAT32, FLOAT16 |
DML_FEATURE_LEVEL_2_0 et versions ultérieures
| Tenseur | Type | Nombre de dimensions pris en charge | Types de données pris en charge |
|---|---|---|---|
| InputTensor | Entrée | 4 à 5 | FLOAT32, FLOAT16 |
| OutputTensor | Output | 4 à 5 | FLOAT32, FLOAT16 |
DML_FEATURE_LEVEL_1_0 et versions ultérieures
| Tenseur | Type | Nombre de dimensions pris en charge | Types de données pris en charge |
|---|---|---|---|
| InputTensor | Entrée | 4 | FLOAT32, FLOAT16 |
| OutputTensor | Output | 4 | FLOAT32, FLOAT16 |
Spécifications
| En-tête | directml.h |