DML_ACTIVATION_SOFTMAX_OPERATOR_DESC-Struktur (directml.h)

Führt eine Softmax-Aktivierungsfunktion für InputTensor aus, und platziert das Ergebnis im entsprechenden Element von OutputTensor.

For 1-D InputTensor:
// Let x[i] 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]))

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

Syntax

struct DML_ACTIVATION_SOFTMAX_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
};

Member

InputTensor

Typ: const DML_TENSOR_DESC*

Der Eingabe tensor, aus dem gelesen werden soll. Dieser Tensor muss einen effektiven Rang haben, der nicht größer als 2 ist. Der effektive Rang eines Tensors ist der DimensionCount des Tensors, wobei die meisten linken Dimensionen der Größe 1 ausgenommen sind. Beispielsweise ist eine Tensorgröße von { 1, 1, BatchCount, Width } gültig und entspricht einem Tensor von Größen { BatchCount, Width }.

OutputTensor

Typ: const DML_TENSOR_DESC*

Der Ausgabe tensor, in den die Ergebnisse geschrieben werden sollen.

Verfügbarkeit

Dieser Operator wurde in DML_FEATURE_LEVEL_1_0eingefü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_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

DML_FEATURE_LEVEL_1_0 und höher

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

Anforderungen

Anforderung	Wert
Header	directml.h

Weitere Informationen

DML_ACTIVATION_LOG_SOFTMAX_OPERATOR_DESC Struktur

DML_ARGMAX_OPERATOR_DESC Struktur