Freigeben über


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

Weitere Informationen

DML_ARGMAX_OPERATOR_DESC-Struktur