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 |