Structure DML_ACTIVATION_HARD_SWISH_OPERATOR_DESC (directml.h)

Article
03/18/2024

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

f(x) = x * HardSigmoid(x, Alpha, Beta)

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.

Important

Cette API est disponible dans le cadre du package redistribuable autonome DirectML (voir Microsoft.AI.DirectML version 1.13 et ultérieures). Consultez également l’historique des versions DirectML.

Syntaxe

struct DML_ACTIVATION_HARD_SWISH_OPERATOR_DESC
{
    const DML_TENSOR_DESC* InputTensor;
    const DML_TENSOR_DESC* OutputTensor;
    FLOAT Alpha;
    FLOAT Beta;
};

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.

Alpha

Type : FLOAT

Coefficient alpha. La valeur par défaut typique est 0,2.

Beta

Type : FLOAT

Coefficient bêta.

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

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

Partager via