structure DML_ELEMENT_WISE_IS_INFINITY_OPERATOR_DESC (directml.h)

Article
11/12/2024

Vérifie chaque élément de InputTensor pour IEEE-754 -inf, inf ou les deux, selon le InfinityModedonné, et place le résultat (1 pour true, 0 pour false) dans l’élément correspondant de OutputTensor.

f(x) = isinf(x) && (
       (x > 0 && InfinityMode == DML_IS_INFINITY_MODE_POSITIVE) ||
       (x < 0 && InfinityMode == DML_IS_INFINITY_MODE_NEGATIVE) ||
                 InfinityMode == DML_IS_INFINITY_MODE_EITHER)

Syntaxe

struct DML_ELEMENT_WISE_IS_INFINITY_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
  DML_IS_INFINITY_MODE  InfinityMode;
};

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.

InfinityMode

Type : DML_IS_INFINITY_MODE

Un DML_IS_INFINITY_MODE déterminant le signe de l’infini à vérifier.

Si DML_IS_INFINITY_MODE_EITHER, 1 est retourné si l’élément est -inf ou inf, sinon 0.
Si DML_IS_INFINITY_MODE_POSITIVE, 1 est retourné si l’élément est inf, sinon 0.
Si DML_IS_INFINITY_MODE_NEGATIVE', 1 est retourné si l’élément est -inf, sinon 0.

Remarques

Disponibilité

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

Contraintes Tensor

InputTensor et OutputTensor doivent avoir le même DimensionCount et 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	UINT8

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	UINT8

Exigences

Exigence	Valeur
client minimum pris en charge	Windows 10, version 2004 (10.0 ; Build 19041)
serveur minimum pris en charge	Windows Server, version 2004 (10.0 ; Build 19041)
d’en-tête	directml.h

Partager via