DML_ACTIVATION_HARDMAX_OPERATOR_DESC-Struktur (directml.h)
Führt eine hardmax-Funktion für jedes Element von InputTensor aus, wobei das Ergebnis im entsprechenden Element von OutputTensor platziert wird.
Der Operator berechnet die hardmax (1 für das erste Vorkommen des größten Werts in der Ebene und 0 für alle anderen Werte) jeder Zeile in der angegebenen Eingabe.
Syntax
struct DML_ACTIVATION_HARDMAX_OPERATOR_DESC {
const DML_TENSOR_DESC *InputTensor;
const DML_TENSOR_DESC *OutputTensor;
};
Member
InputTensor
Typ: const DML_TENSOR_DESC*
Der Tensor, aus dem für die Eingabe gelesen werden soll. Dieser Tensor muss einen effektiven Rang haben, der nicht größer als 2 ist. Der effektive Rang eines Tensors ist die DimensionCount des Tensors, mit Ausnahme der äußersten linken Dimensionen der Größe 1. Beispielsweise ist eine Tensorgröße von { 1, 1, BatchCount, Width } gültig und entspricht einem Tensor der Größen { BatchCount, Width }.
OutputTensor
Typ: const DML_TENSOR_DESC*
Der Ausgabe tensor, in den die Ergebnisse geschrieben werden sollen.
Hinweise
Der Operator berechnet die Werte hardmax (1 für den ersten Höchstwert und 0 für alle anderen) für jede Ebene im Batch der angegebenen Eingabe. Die Eingabe ist ein 2D-Tensor (Tensor) der Größe (batch_size x input_feature_dimensions). Der Ausgabe tensor hat die gleiche Form und enthält die hardmax-Werte der entsprechenden Eingabe.
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 |