Freigeben über

DML_ELEMENT_WISE_POW_OPERATOR_DESC-Struktur (directml.h)

  • Artikel

Berechnet jedes Element von InputTensor , das auf die Leistung des entsprechenden Elements von ExponentTensor angehoben wird, und platziert das Ergebnis in das entsprechende Element von OutputTensor.

f(input, exponent) = pow(input, exponent)

Negative Basen werden für Exponenten mit integralen Werten unterstützt (obwohl der Datentyp weiterhin float sein kann), andernfalls gibt dieser Operator NaN zurück.

Wenn der Eingabe tensor und der Exponentensor beide über einen integralen Datentyp verfügen, garantiert dieser Operator exakte Ergebnisse.

Dieser Operator unterstützt die direkte Ausführung, was bedeutet, dass OutputTensor während der Bindung den Alias InputTensor verwenden darf.

Syntax

struct DML_ELEMENT_WISE_POW_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *ExponentTensor;
  const DML_TENSOR_DESC *OutputTensor;
  const DML_SCALE_BIAS  *ScaleBias;
};

Member

InputTensor

Typ: const DML_TENSOR_DESC*

Der Tensor, der die Eingabewerte enthält.

ExponentTensor

Typ: const DML_TENSOR_DESC*

Der Tensor, der die Exponentenwerte enthält.

OutputTensor

Typ: const DML_TENSOR_DESC*

Der Ausgabe tensor, in den die Ergebnisse geschrieben werden sollen.

ScaleBias

Typ: _Maybenull_ const DML_SCALE_BIAS*

Eine optionale Skalierung und Verzerrung, die auf die Eingabe angewendet werden soll. Falls vorhanden, bewirkt dies, dass die Funktion g(x) = x * scale + bias vor dem Berechnen dieses Operators auf jedes Eingabeelement angewendet wird.

Hinweise

Bis DML_FEATURE_LEVEL_3_0mussexponentTensor mit dem Typ von InputTensor übereinstimmen.

Siehe DML_ELEMENT_WISE_CONSTANT_POW_OPERATOR_DESC für einen POW-Operator, der eine FLOAT Konstante für den Exponenten akzeptiert.

Verfügbarkeit

Dieser Operator wurde in DML_FEATURE_LEVEL_1_0eingeführt.

Tensoreinschränkungen

  • ExponentTensor, InputTensor und OutputTensor müssen über die gleichen DimensionCount und Sizes verfügen.
  • InputTensor und OutputTensor müssen denselben Datentyp 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, INT32, INT16, INT8, UINT32, UINT16, UINT8
ExponentTensor Eingabe 1 bis 8 FLOAT32, FLOAT16, INT32, INT16, INT8, UINT32, UINT16, UINT8
OutputTensor Ausgabe 1 bis 8 FLOAT32, FLOAT16, INT32, INT16, INT8, UINT32, UINT16, UINT8

DML_FEATURE_LEVEL_1_0 und höher

Tensor Typ Unterstützte Dimensionsanzahl Unterstützte Datentypen
InputTensor Eingabe 4 FLOAT32, FLOAT16
ExponentTensor Eingabe 4 FLOAT32, FLOAT16
OutputTensor Ausgabe 4 FLOAT32, FLOAT16

Anforderungen

   
Kopfzeile directml.h

Weitere Informationen

DML_ELEMENT_WISE_CONSTANT_POW_OPERATOR_DESC