Partager via


structure DML_ELEMENT_WISE_IS_INFINITY_OPERATOR_DESC (directml.h)

Vérifie chaque élément d’InputTensor pour IEEE-754 -inf, inf ou les deux, en fonction de l’InfinityMode donné, 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*

Tenseur d’entrée à partir duquel lire.

OutputTensor

Type : const DML_TENSOR_DESC*

Tenseur 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 pour lequel case activée.

  • 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.

Notes

Disponibilité

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

Contraintes tensoriels

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

Prise en charge de Tensor

DML_FEATURE_LEVEL_3_0 et versions ultérieures

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

DML_FEATURE_LEVEL_2_1 et versions ultérieures

Tenseur Type Nombre de dimensions pris en charge Types de données pris en charge
InputTensor Entrée 4 FLOAT32, FLOAT16
OutputTensor Output 4 UINT8

Spécifications

   
Client minimal pris en charge Windows 10, version 2004 (10.0 ; Build 19041)
Serveur minimal pris en charge Windows Server, version 2004 (10.0 ; Build 19041)
En-tête directml.h