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 |