DML_ELEMENT_WISE_ROUND_OPERATOR_DESC structure (directml.h)

Article
02/26/2024

Arrondit chaque élément de InputTensor à une valeur entière, en plaçant le résultat dans l’élément correspondant de OutputTensor.

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

Syntaxe

struct DML_ELEMENT_WISE_ROUND_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
  DML_ROUNDING_MODE     RoundingMode;
};

Membres

InputTensor

Type : const DML_TENSOR_DESC*

Tenseur d’entrée à partir duquel lire.

OutputTensor

Type : const DML_TENSOR_DESC*

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

RoundingMode

Type : DML_ROUNDING_MODE

Un DML_ROUNDING_MODE déterminant la direction vers laquelle arrondir.

Si DML_ROUNDING_MODE_HALVES_TO_NEAREST_EVEN : les valeurs sont arrondies à l’entier le plus proche, avec des valeurs à mi-chemin (par exemple, 0,5) arrondies à l’entier pair le plus proche.
Si DML_ROUNDING_MODE_TOWARD_ZERO : les valeurs sont arrondies à zéro. Cela tronque effectivement la partie fractionnaire.
Si DML_ROUNDING_MODE_TOWARD_INFINITY : les valeurs sont arrondies à l’entier le plus proche, les valeurs à mi-chemin (par exemple, 0,5) étant arrondies de zéro (vers l’infini positif ou négatif, selon le signe de la valeur).

Disponibilité

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

Contraintes tensoriels

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

Prise en charge des tenseurs

DML_FEATURE_LEVEL_3_0 et versions ultérieures

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

DML_FEATURE_LEVEL_2_1 et versions ultérieures

Tenseur	Genre	Nombre de dimensions pris en charge	Types de données pris en charge
InputTensor	Entrée	4	FLOAT32, FLOAT16
OutputTensor	Sortie	4	FLOAT32, FLOAT16

Configuration requise

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

Partager via