Partager via

DML_ELEMENT_WISE_ADD_OPERATOR_DESC structure (directml.h)

  • Article

Ajoute chaque élément dans ATensor à son élément correspondant dans BTensor, en plaçant le résultat dans l’élément correspondant de OutputTensor.

f(a, b) = a + b

Cet opérateur prend en charge l’exécution sur place, ce qui signifie que OutputTensor est autorisé à alias un ou plusieurs des tenseurs d’entrée pendant la liaison.

Syntaxe

struct DML_ELEMENT_WISE_ADD_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 du côté gauche.

BTensor

Type : const DML_TENSOR_DESC*

Tenseur contenant les entrées du côté droit.

OutputTensor

Type : const DML_TENSOR_DESC*

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

Notes

Une version plus récente de cet opérateur, DML_ELEMENT_WISE_ADD1_OPERATOR_DESC, a été introduite dans DML_FEATURE_LEVEL_2_0.

Disponibilité

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

Contraintes tensoriels

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

Prise en charge des tenseurs

DML_FEATURE_LEVEL_5_1 et versions ultérieures

Tenseur Type Nombre de dimensions pris en charge Types de données pris en charge
ATensor Entrée 1 à 8 FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8
BTensor Entrée 1 à 8 FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8
OutputTensor Output 1 à 8 FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8

DML_FEATURE_LEVEL_4_1 et versions ultérieures

Tenseur Type Nombre de dimensions pris en charge Types de données pris en charge
ATensor Entrée 1 à 8 FLOAT32, FLOAT16, INT64, INT32, UINT64, UINT32
BTensor Entrée 1 à 8 FLOAT32, FLOAT16, INT64, INT32, UINT64, UINT32
OutputTensor Output 1 à 8 FLOAT32, FLOAT16, INT64, INT32, UINT64, UINT32

DML_FEATURE_LEVEL_3_0 et versions ultérieures

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

DML_FEATURE_LEVEL_2_1 et versions ultérieures

Tenseur Type Nombre de dimensions pris en charge Types de données pris en charge
ATensor Entrée 4 à 5 FLOAT32, FLOAT16, INT32, UINT32
BTensor Entrée 4 à 5 FLOAT32, FLOAT16, INT32, UINT32
OutputTensor Output 4 à 5 FLOAT32, FLOAT16, INT32, UINT32

DML_FEATURE_LEVEL_1_0 et versions ultérieures

Tenseur Type Nombre de dimensions pris en charge Types de données pris en charge
ATensor Entrée 4 à 5 FLOAT32, FLOAT16
BTensor Entrée 4 à 5 FLOAT32, FLOAT16
OutputTensor Output 4 à 5 FLOAT32, FLOAT16

Spécifications

   
En-tête directml.h

Voir aussi

DML_ELEMENT_WISE_ADD1_OPERATOR_DESC