structure DML_VALUE_SCALE_2D_OPERATOR_DESC (directml.h)

Exécute une fonction de mise à l’échelle et de biais au niveau des éléments, Output = Scale * Input + Bias. Cet opérateur est similaire à l’utilisation d’un DML_ELEMENT_WISE_IDENTITY_OPERATOR_DESC avec une échelle et un biais, sauf que DML_VALUE_SCALE_2D_OPERATOR_DESC applique un biais différent pour chaque canal, plutôt qu’un biais unique pour l’ensemble du tenseur.

Syntaxe

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

Membres

InputTensor

Type : const DML_TENSOR_DESC*

Tenseur contenant les données d’entrée. Les dimensions de ce tenseur doivent être { BatchCount, ChannelCount, Height, Width }.

OutputTensor

Type : const DML_TENSOR_DESC*

Tenseur dans lequel écrire les résultats. Les dimensions de ce tenseur doivent correspondre aux dimensions du capteur InputTensor.

Scale

Type : FLOAT

Valeur d’échelle à appliquer à toutes les valeurs d’entrée.

ChannelCount

Type : UINT

Ce champ détermine la taille du tableau Bias. Ce champ doit être défini sur 1 ou 3, et doit également correspondre à la taille de la dimension Channel du tenseur d’entrée.

Bias

Type : const FLOAT*

Tableau de valeurs FLOAT contenant le terme de biais pour chaque dimension du tenseur d’entrée.

Disponibilité

Cet opérateur a été introduit dans DML_FEATURE_LEVEL_1_0.

Contraintes tensoriels

InputTensor et OutputTensor doivent avoir les mêmes DataType et Sizes.

Prise en charge des tenseurs

Configuration requise