Partager via


structure DML_ELEMENT_WISE_DIFFERENCE_SQUARE_OPERATOR_DESC (directml.h)

Soustrait chaque élément de BTensor de l’élément correspondant d’ATensor, multiplie le résultat par lui-même et place le résultat dans l’élément correspondant de OutputTensor.

f(a, b) = (a - b) * (a - b)

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

Syntaxe

struct DML_ELEMENT_WISE_DIFFERENCE_SQUARE_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 contenant les entrées de gauche.

BTensor

Type : const DML_TENSOR_DESC*

Tenseur contenant les entrées de droite.

OutputTensor

Type : const DML_TENSOR_DESC*

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

Notes

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 Type Nombre de dimensions pris en charge Types de données pris en charge
ATensor Entrée 1 à 8 FLOAT32, FLOAT16, INT32, UINT32
BTensor Entrée 1 à 8 FLOAT32, FLOAT16, INT32, UINT32
OutputTensor Output 1 à 8 FLOAT32, FLOAT16, INT32, UINT32

Spécifications

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