DML_ACTIVATION_SWISH_OPERATOR_DESC-Struktur (directml.h)

Artikel
03/18/2024

Führt eine Swish-Aktivierungsfunktion für jedes Element in InputTensor aus, wobei das Ergebnis in das entsprechende Element von OutputTensor eingefügt wird.

f(x) = x * Sigmoid(SigmoidInputScale * x)

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

Wichtig

Diese API ist als Teil des eigenständigen weiterverteilbare Pakets DirectML verfügbar (siehe Microsoft.AI.DirectML , Version 1.13 und höher). Siehe auch DirectML-Versionsverlauf.

Syntax

struct DML_ACTIVATION_SWISH_OPERATOR_DESC
{
    const DML_TENSOR_DESC* InputTensor;
    const DML_TENSOR_DESC* OutputTensor;
    FLOAT SigmoidInputScale;
};

Member

InputTensor

Typ: const DML_TENSOR_DESC*

Der Eingabe-Tensor, aus dem gelesen wird.

OutputTensor

Typ: const DML_TENSOR_DESC*

Der Ausgabe-Tensor, in die das Ergebnis geschrieben werden soll.

SigmoidInputScale

Typ: FLOAT

Der Konstantenkoeffizienten. Ein typischer Standardwert für diesen Wert ist 1.0.

Verfügbarkeit

Dieser Operator wurde in DML_FEATURE_LEVEL_6_2 eingeführt.

Tensor-Einschränkungen

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

Tensor-Unterstützung

Tensor	Variante	Unterstützte Dimensionsanzahl	Unterstützte Datentypen
InputTensor	Eingabe	1 bis 8	FLOAT32, FLOAT16
OutputTensor	Output	1 bis 8	FLOAT32, FLOAT16

Anforderungen


Übergeordnet	directml.h

Freigeben über