DML_ACTIVATION_LOG_SOFTMAX_OPERATOR_DESC-Struktur (directml.h)
Führt eine natürliche Log-of-Softmax-Aktivierungsfunktion für jedes Element von InputTensor aus, und platziert das Ergebnis im entsprechenden Element von OutputTensor.
// Let x_i be the current value in the axis, and j be the total number of elements along that axis.
f(x_i) = ln(exp(x_i) / sum(exp(x_0), ..., exp(x_j)))
Wobei exp(x) die natürliche Exponentiationsfunktion und ln(x) der natürliche Logarithmus ist.
Syntax
struct DML_ACTIVATION_LOG_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 | Typ | 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 | Typ | 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 | Typ | Unterstützte Dimensionsanzahl | Unterstützte Datentypen |
|---|---|---|---|
| InputTensor | Eingabe | 4 | FLOAT32, FLOAT16 |
| OutputTensor | Ausgabe | 4 | FLOAT32, FLOAT16 |
Anforderungen
| Kopfzeile | directml.h |