DML_ACTIVATION_SOFTMAX1_OPERATOR_DESC Struktur (directml.h)

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

For 1-D InputTensor:
// Let x[i] to be the current element in the InputTensor, and j be the total number of elements in the InputTensor
f(x[i]) = exp(x[i]) / sum(exp(x[0]), ..., exp(x[j-1]))

Dabei ist exp(x) die natürliche Exponentiationsfunktion.

Syntax

struct DML_ACTIVATION_SOFTMAX1_OPERATOR_DESC
{
    const DML_TENSOR_DESC* InputTensor;
    const DML_TENSOR_DESC* OutputTensor;
    UINT AxisCount;
    _Field_size_(AxisCount) const UINT* Axes;
};

Elemente

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.

AxisCount

Typ: UINT

Die Anzahl der Achsen, die berechnet werden sollen, um die Summe zu reduzieren. Dieses Feld bestimmt die Größe des Axes-Arrays .

Axes

Typ: _Field_size_(AxisCount) const UINT*

Die Achsen, entlang der die Summe reduziert werden soll. Werte müssen sich im Bereich [0, InputTensor.DimensionCount - 1]befinden.

Bemerkungen

Dieser Operator entspricht DML_ACTIVATION_SOFTMAX_OPERATOR_DESC , wenn AxisCount == 1 und Achsen == {DimensionCount - 1} .

Verfügbarkeit

Dieser Operator wurde in DML_FEATURE_LEVEL_5_1 eingeführt.

Tensoreinschränkungen

"InputTensor " und " OutputTensor " müssen denselben Datentyp, "DimensionCount" und " Größen" aufweisen.

Tensorunterstützung

Anforderungen

Siehe auch

• DML_ACTIVATION_LOG_SOFTMAX1_OPERATOR_DESC