DML_VALUE_SCALE_2D_OPERATOR_DESC結構 (directml.h)

執行以元素為基礎的縮放和偏差函式 。 Output = Scale * Input + Bias 此運算符類似於使用具有縮放比例和偏差 的DML_ELEMENT_WISE_IDENTITY_OPERATOR_DESC ，不同之處在於 DML_VALUE_SCALE_2D_OPERATOR_DESC 針對每個通道套用不同的偏差，而不是整個張量的單一偏差。

語法

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

成員

InputTensor

類型： const DML_TENSOR_DESC*

包含輸入數據的張量。 這個張量維度應該是 { BatchCount, ChannelCount, Height, Width }。

OutputTensor

類型： const DML_TENSOR_DESC*

要寫入結果的張量。 此張量維度應該符合 InputTensor 的維度。

Scale

類型： FLOAT

要套用至所有輸入值的縮放值。

ChannelCount

類型： UINT

此欄位會決定 Bias 陣列的大小。 此欄位必須設定為 1 或 3，而且也必須符合輸入張量之 Channel 維度的大小。

Bias

類型： const FLOAT*

FLOAT 值的陣列，其中包含輸入張量之每個維度的偏差字詞。

可用性

這個運算子是在 中 DML_FEATURE_LEVEL_1_0引進。

Tensor 條件約束

InputTensor 和 OutputTensor 必須具有相同 的 DataType 和 大小。

Tensor 支援

張	種類	支援的維度計數	支援的資料類型
InputTensor	輸入	4	FLOAT32，FLOAT16
OutputTensor	輸出	4	FLOAT32，FLOAT16

規格需求

需求	值
標頭	directml.h