DML_VALUE_SCALE_2D_OPERATOR_DESC-Struktur (directml.h)

Artikel
02/26/2024

Führt eine elementweise Skalierungs- und Biasfunktion aus, Output = Scale * Input + Bias. Dieser Operator ähnelt der Verwendung eines DML_ELEMENT_WISE_IDENTITY_OPERATOR_DESC mit einer Skalierung und Verzerrung, mit der Ausnahme, dass DML_VALUE_SCALE_2D_OPERATOR_DESC eine andere Verzerrung für jeden Kanal anwendet, anstatt eine einzelne Verzerrung für den gesamten Tensor.

Syntax

struct DML_VALUE_SCALE_2D_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
  FLOAT                 Scale;
  UINT                  ChannelCount;
  const FLOAT           *Bias;
};

Member

InputTensor

Typ: const DML_TENSOR_DESC*

Ein Tensor, der die Eingabedaten enthält. Die Dimensionen dieses Tensors sollten sein { BatchCount, ChannelCount, Height, Width }.

OutputTensor

Typ: const DML_TENSOR_DESC*

Ein Tensor, mit dem die Ergebnisse geschrieben werden sollen. Die Dimensionen dieses Tensors sollten mit den Dimensionen des InputTensors übereinstimmen.

Scale

Typ: FLOAT

Skalierungswert, der auf alle Eingabewerte angewendet werden soll.

ChannelCount

Typ: UINT

Dieses Feld bestimmt die Größe des Bias-Arrays. Dieses Feld muss auf 1 oder 3 festgelegt werden und muss auch der Größe der Kanaldimension des Eingabe tensors entsprechen.

Bias

Typ: const FLOAT*

Ein Array von FLOAT-Werten , das den Biasbegriff für jede Dimension des Eingabe tensors enthält.

Verfügbarkeit

Dieser Operator wurde in DML_FEATURE_LEVEL_1_0eingeführt.

Tensoreinschränkungen

InputTensor und OutputTensor müssen denselben Datentyp und dieselben Größen aufweisen.

Tensorunterstützung

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

Anforderungen

Anforderung	Wert
Header	directml.h

Freigeben über