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 |