structure DML_ELEMENT_WISE_ROUND_OPERATOR_DESC (directml.h)

Article
11/12/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*

Capteur d’entrée à partir duquel lire.

OutputTensor

Type : const DML_TENSOR_DESC*

Le tensoriel de sortie dans lequel écrire les résultats.

RoundingMode

Type : DML_ROUNDING_MODE

Une 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 vers l’entier pair le plus proche.
Si DML_ROUNDING_MODE_TOWARD_ZERO: les valeurs sont arrondies à zéro. Cela tronque efficacement la partie fractionnaire.
Si DML_ROUNDING_MODE_TOWARD_INFINITY: les valeurs sont arrondies à l’entier le plus proche, avec des valeurs de moitié (par exemple, 0,5) arrondies à 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 Tensor

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

Prise en charge de Tensor

DML_FEATURE_LEVEL_3_0 et versions ultérieures

Tenseur	Gentil	Nombres 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	Gentil	Nombres de dimensions pris en charge	Types de données pris en charge
InputTensor	Entrée	4	FLOAT32, FLOAT16
OutputTensor	Sortie	4	FLOAT32, FLOAT16

Exigences

Exigence	Valeur
client minimum pris en charge	Windows 10 Build 20348
serveur minimum pris en charge	Windows 10 Build 20348
d’en-tête	directml.h

Partager via