DML_ACTIVATION_SHRINK_OPERATOR_DESC-Struktur (directml.h)

Artikel
03/01/2024

Führt die Verkleinerungsaktivierungsfunktion für jedes Element in InputTensor aus, und platziert das Ergebnis in das entsprechende Element von OutputTensor.

f(x) = x - Bias, if x > Threshold
       x + Bias, if x < -Threshold
       0,        otherwise

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

Syntax

struct DML_ACTIVATION_SHRINK_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
  FLOAT                 Bias;
  FLOAT                 Threshold;
};

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.

Bias

Typ: FLOAT

Der Wert der Verzerrung. Ein typischer Standardwert für diesen Wert ist 0,0.

Threshold

Typ: FLOAT

Der Wert des Schwellenwerts. Ein typischer Standardwert für diesen Wert ist 0,5.

Hinweise

Verfügbarkeit

Dieser Operator wurde in DML_FEATURE_LEVEL_2_0eingeführt.

Tensoreinschränkungen

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

Tensorunterstützung

DML_FEATURE_LEVEL_5_1 und höher

Tensor	Variante	Unterstützte Dimensionsanzahl	Unterstützte Datentypen
InputTensor	Eingabe	1 bis 8	FLOAT32, FLOAT16, INT32, INT16, INT8, UINT32, UINT16, UINT8
OutputTensor	Ausgabe	1 bis 8	FLOAT32, FLOAT16, INT32, INT16, INT8, UINT32, UINT16, UINT8

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_0 und höher

Tensor	Variante	Unterstützte Dimensionsanzahl	Unterstützte Datentypen
InputTensor	Eingabe	4 bis 5	FLOAT32, FLOAT16
OutputTensor	Ausgabe	4 bis 5	FLOAT32, FLOAT16

Anforderungen

Anforderung	Wert
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

Freigeben über