Share via


DML_ELEMENT_WISE_IS_INFINITY_OPERATOR_DESC-Struktur (directml.h)

Überprüft jedes Element von InputTensor auf IEEE-754 -inf, inf oder beides, abhängig vom angegebenen InfinityMode, und platziert das Ergebnis (1 für true, 0 für false) in das entsprechende Element von 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)

Syntax

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

Member

InputTensor

Typ: const DML_TENSOR_DESC*

Der Eingabe tensor, aus dem gelesen werden soll.

OutputTensor

Typ: const DML_TENSOR_DESC*

Der Ausgabe tensor, in den die Ergebnisse geschrieben werden sollen.

InfinityMode

Typ: DML_IS_INFINITY_MODE

Ein DML_IS_INFINITY_MODE , der das Zu überprüfende Zeichen der Unendlichkeit bestimmt.

  • Wenn DML_IS_INFINITY_MODE_EITHER, wird 1 zurückgegeben, wenn das Element -inf oder inf ist, andernfalls 0.
  • Wenn DML_IS_INFINITY_MODE_POSITIVE, wird 1 zurückgegeben, wenn das Element inf ist, andernfalls 0.
  • Wenn DML_IS_INFINITY_MODE_NEGATIVE, wird 1 zurückgegeben, wenn das Element -inf ist, andernfalls 0.

Hinweise

Verfügbarkeit

Dieser Operator wurde in DML_FEATURE_LEVEL_2_1eingeführt.

Tensoreinschränkungen

InputTensor und OutputTensor müssen über die gleichen DimensionCount und Sizes verfügen.

Tensorunterstützung

DML_FEATURE_LEVEL_3_0 und höher

Tensor Typ Unterstützte Dimensionsanzahl Unterstützte Datentypen
InputTensor Eingabe 1 bis 8 FLOAT32, FLOAT16
OutputTensor Ausgabe 1 bis 8 UINT8

DML_FEATURE_LEVEL_2_1 und höher

Tensor Typ Unterstützte Dimensionsanzahl Unterstützte Datentypen
InputTensor Eingabe 4 FLOAT32, FLOAT16
OutputTensor Ausgabe 4 UINT8

Anforderungen

   
Unterstützte Mindestversion (Client) Windows 10, Version 2004 (10.0; Build 19041)
Unterstützte Mindestversion (Server) Windows Server, Version 2004 (10.0; Build 19041)
Kopfzeile directml.h