Structure DML_ACTIVATION_SWISH_OPERATOR_DESC (directml.h)

Exécute une fonction swish sur chaque élément dans InputTensor, en plaçant le résultat dans l’élément OutputTensor correspondant.

f(x) = x * Sigmoid(SigmoidInputScale * x)

Cet opérateur prend en charge l’exécution sur place, ce qui signifie que le tenseur de sortie est autorisé à prendre InputTensor en tant qu’alias durant la liaison.

Syntaxe

struct DML_ACTIVATION_SWISH_OPERATOR_DESC
{
    const DML_TENSOR_DESC* InputTensor;
    const DML_TENSOR_DESC* OutputTensor;
    FLOAT SigmoidInputScale;
};

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.

SigmoidInputScale

Type : FLOAT

Coefficient de constante. La valeur par défaut typique est 1.0.

Disponibilité

Cet opérateur a été introduit dans DML_FEATURE_LEVEL_6_2.

Contraintes de tenseur

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

Prise en charge des tenseurs

Spécifications