Partager via


structure DML_ELEMENT_WISE_ATAN_YX_OPERATOR_DESC (directml.h)

Calcule l’arctangent à 2 arguments pour chaque élément d’ATensor et BTensor, où ATensor est l’axe Y et BTensor est l’axe X, plaçant le résultat dans l’élément outputTensor correspondant. Cet opérateur n’est pas défini pour l’origine (autrement dit, lorsque ATensor et BTensor sont tous deux 0 pour les éléments correspondants).

GRU_Forward

f(y, x) = atan2(y, x)

Cet opérateur prend en charge l’exécution sur place, ce qui signifie que le tenseur de sortie est autorisé à alias ATensor ou BTensor pendant la liaison.

Syntaxe

struct DML_ELEMENT_WISE_ATAN_YX_OPERATOR_DESC {
  const DML_TENSOR_DESC *ATensor;
  const DML_TENSOR_DESC *BTensor;
  const DML_TENSOR_DESC *OutputTensor;
};

Membres

ATensor

Type : const DML_TENSOR_DESC*

Tenseur d’entrée à partir duquel lire les valeurs de l’axe Y.

BTensor

Type : const DML_TENSOR_DESC*

Tenseur d’entrée à partir duquel lire les valeurs de l’axe X.

OutputTensor

Type : const DML_TENSOR_DESC*

Tenseur de sortie dans lequel écrire les résultats.

Remarques

Disponibilité

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

Contraintes tensoriels

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

Prise en charge de Tensor

Tenseur Genre Nombre de dimensions pris en charge Types de données pris en charge
ATensor Entrée 1 à 8 FLOAT32, FLOAT16
BTensor Entrée 1 à 8 FLOAT32, FLOAT16
OutputTensor Sortie 1 à 8 FLOAT32, FLOAT16

Configuration requise

Condition requise Valeur
Client minimal pris en charge Windows Build 22000
Serveur minimal pris en charge Windows Build 22000
En-tête directml.h