DML_ELEMENT_WISE_ROUND_OPERATOR_DESC-Struktur (directml.h)

Artikel
02/26/2024

Rundet jedes Element von InputTensor auf einen ganzzahligen Wert ab, und platziert das Ergebnis im entsprechenden Element von OutputTensor.

Dieser Operator unterstützt die direkte Ausführung, was bedeutet, dass OutputTensor während der Bindung den Alias InputTensor verwenden darf.

Syntax

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

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.

RoundingMode

Typ: DML_ROUNDING_MODE

Eine DML_ROUNDING_MODE , die die Richtung bestimmt, in die sich eine Runde drehen soll.

Wenn DML_ROUNDING_MODE_HALVES_TO_NEAREST_EVEN: Die Werte werden auf die nächste ganze Zahl gerundet, wobei die Halbwertwerte (z. B. 0,5) auf die nächste gerade ganze Zahl gerundet werden.
Wenn DML_ROUNDING_MODE_TOWARD_ZERO: Die Werte werden auf 0 gerundet. Dadurch wird der Bruchteil effektiv abgeschnitten.
Wenn DML_ROUNDING_MODE_TOWARD_INFINITY: Werte werden auf die nächste ganze Zahl gerundet, wobei halbwegswerte (z. B. 0,5) von 0 abgerundet werden (je nach Vorzeichen des Werts in Richtung positive oder negative Unendlichkeit).

Verfügbarkeit

Dieser Operator wurde in DML_FEATURE_LEVEL_2_1eingeführt.

Tensoreinschränkungen

InputTensor und OutputTensor müssen denselben DataType, DimensionCount und Sizes aufweisen.

Tensorunterstützung

DML_FEATURE_LEVEL_3_0 und höher

Tensor	Variante	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	Variante	Unterstützte Dimensionsanzahl	Unterstützte Datentypen
InputTensor	Eingabe	4	FLOAT32, FLOAT16
OutputTensor	Ausgabe	4	FLOAT32, FLOAT16

Anforderungen

Anforderung	Wert
Unterstützte Mindestversion (Client)	Windows 10 Build 20348
Unterstützte Mindestversion (Server)	Windows 10 Build 20348
Kopfzeile	directml.h

Share via