structure DML_ELEMENT_WISE_MULTIPLY_OPERATOR_DESC (directml.h)
Calcule le produit de chaque paire d’éléments correspondants des tenseurs d’entrée, en plaçant le résultat dans l’élément OutputTensor correspondant.
f(a, b) = a * b
Cet opérateur prend en charge l’exécution sur place, ce qui signifie que OutputTensor est autorisé à alias l’un des tenseurs d’entrée pendant la liaison.
Syntaxe
struct DML_ELEMENT_WISE_MULTIPLY_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.
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 de Tensor
DML_FEATURE_LEVEL_5_1 et versions ultérieures
Tenseur | Genre | 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 | Sortie | 1 à 8 | FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8 |
DML_FEATURE_LEVEL_4_1 et versions ultérieures
Tenseur | Genre | 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 | Sortie | 1 à 8 | FLOAT32, FLOAT16, INT64, INT32, UINT64, UINT32 |
DML_FEATURE_LEVEL_3_0 et versions ultérieures
Tenseur | Genre | 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 | Sortie | 1 à 8 | FLOAT32, FLOAT16, INT32, UINT32 |
DML_FEATURE_LEVEL_2_1 et versions ultérieures
Tenseur | Genre | 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 | Sortie | 4 à 5 | FLOAT32, FLOAT16, INT32, UINT32 |
DML_FEATURE_LEVEL_1_0 et versions ultérieures
Tenseur | Genre | 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 | Sortie | 4 à 5 | FLOAT32, FLOAT16 |
Configuration requise
Condition requise | Valeur |
---|---|
En-tête | directml.h |
Commentaires
https://aka.ms/ContentUserFeedback.
Bientôt disponible : Tout au long de 2024, nous allons supprimer progressivement GitHub Issues comme mécanisme de commentaires pour le contenu et le remplacer par un nouveau système de commentaires. Pour plus d’informations, consultezEnvoyer et afficher des commentaires pour