DML_ELEMENT_WISE_ROUND_OPERATOR_DESC Struktur (directml.h)

Artikel
11/12/2024

Rundet jedes Element von InputTensor auf einen ganzzahligen Wert ab und platziert das Ergebnis in das entsprechende Element OutputTensor.

Dieser Operator unterstützt die direkte Ausführung, was bedeutet, dass OutputTensor- während der Bindung alias InputTensor ist.

Syntax

struct DML_ELEMENT_WISE_ROUND_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
  DML_ROUNDING_MODE     RoundingMode;
};

Angehörige

InputTensor

Typ: DML_TENSOR_DESC*

Der Eingabe-Tensor, aus dem gelesen werden soll.

OutputTensor

Typ: DML_TENSOR_DESC*

Der Ausgabe-Tensor, in den die Ergebnisse geschrieben werden sollen.

RoundingMode

Typ: DML_ROUNDING_MODE

Ein DML_ROUNDING_MODE die Richtung bestimmen, die in Richtung gerundet werden soll.

Wenn DML_ROUNDING_MODE_HALVES_TO_NEAREST_EVEN: Werte auf die nächste ganze Zahl gerundet werden, wobei halbwegswerte (z. B. 0,5) auf die nächste gerade ganze Zahl gerundet werden.
Wenn DML_ROUNDING_MODE_TOWARD_ZERO: Werte auf Null gerundet werden. Dadurch wird der Bruchteil effektiv abgeschnitten.
Wenn DML_ROUNDING_MODE_TOWARD_INFINITY: Werte auf die nächste ganze Zahl gerundet werden, wobei halbwegswerte (z. B. 0,5) von Null (in Richtung positiver oder negativer Unendlichkeit, abhängig vom Vorzeichen des Werts) abgerundet werden.

Verfügbarkeit

Dieser Operator wurde in DML_FEATURE_LEVEL_2_1eingeführt.

Tensoreinschränkungen

InputTensor- und OutputTensor- müssen denselben Datentyp, DimensionCountund Größenhaben.

Tensorunterstützung

DML_FEATURE_LEVEL_3_0 und höher

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

DML_FEATURE_LEVEL_2_1 und höher

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

Anforderungen

Anforderung	Wert
mindestens unterstützte Client-	Windows 10 Build 20348
mindestens unterstützte Server-	Windows 10 Build 20348
Header-	directml.h

Freigeben über