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

Weitere Informationen

DML_ACTIVATION_SOFTMAX_OPERATOR_DESC Struktur

DML_ARGMAX_OPERATOR_DESC Struktur