Condividi tramite


DML_VALUE_SCALE_2D_OPERATOR_DESC struttura (directml.h)

Esegue una funzione scale-and-bias a livello di elemento, Output = Scale * Input + Bias. Questo operatore è simile all'uso di un DML_ELEMENT_WISE_IDENTITY_OPERATOR_DESC con una scala e un pregiudizio, ad eccezione del fatto che DML_VALUE_SCALE_2D_OPERATOR_DESC applica un pregiudizio diverso per ogni canale, anziché un unico pregiudizio per l'intero tensore.

Sintassi

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

Members

InputTensor

Tipo: const DML_TENSOR_DESC*

Tensore contenente i dati di input. Le dimensioni di questo tensore devono essere { BatchCount, ChannelCount, Height, Width }.

OutputTensor

Tipo: const DML_TENSOR_DESC*

Tensore a cui scrivere i risultati. Le dimensioni del tensore devono corrispondere alle dimensioni di InputTensor.

Scale

Tipo: FLOAT

Valore di scala da applicare a tutti i valori di input.

ChannelCount

Tipo: UINT

Questo campo determina le dimensioni della matrice Bias. Questo campo deve essere impostato su 1 o 3 e deve corrispondere anche alle dimensioni della dimensione Channel del tensore di input.

Bias

Tipo: const FLOAT*

Matrice di valori FLOAT contenenti il termine di pregiudizio per ogni dimensione del tensore di input.

Disponibilità

Questo operatore è stato introdotto in DML_FEATURE_LEVEL_1_0.

Vincoli tensor

InputTensor e OutputTensor devono avere lo stesso tipo di dati e dimensioni.

Supporto di Tensor

Tensore Tipo Conteggi delle dimensioni supportate Tipi di dati supportati
InputTensor Input 4 FLOAT32, FLOAT16
OutputTensor Output 4 FLOAT32, FLOAT16

Requisiti

Requisito Valore
Intestazione directml.h