共用方式為


directml.h (DML_LOCAL_RESPONSE_NORMALIZATION_OPERATOR_DESC 結構)

在輸入上執行本機回應正規化 (LRN) 函式。 這個運算子會執行下列計算。

Output = Input / (Bias + (Alpha / LocalSize) * sum(Input^2 for every Input in the local region))^Beta

輸入和輸出張量的資料型別和大小必須相同。

語法

struct DML_LOCAL_RESPONSE_NORMALIZATION_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
  BOOL                  CrossChannel;
  UINT                  LocalSize;
  FLOAT                 Alpha;
  FLOAT                 Beta;
  FLOAT                 Bias;
};

成員

InputTensor

類型:const DML_TENSOR_DESC*

包含輸入資料的張量。 此張量 的大小 應該是 { BatchCount, ChannelCount, Height, Width }

OutputTensor

類型:const DML_TENSOR_DESC*

要寫入結果的張量。 此張量 的大小 應該符合 InputTensor

CrossChannel

類型: BOOL

如果 LRN 層在通道之間加總,則為TRUE;否則為FALSE

LocalSize

類型: UINT

每個維度要加總的專案數:如果) 已設定 CrossChannel ,則為 Width、Height 和 optionally Channel (。 此值必須至少為 1。

Alpha

類型: FLOAT

縮放參數的值。 建議使用預設值為 0.0001 的值。

Beta

類型: FLOAT

指數值。 建議使用值為 0.75 作為預設值。

Bias

類型: FLOAT

偏差的值。 建議使用值為 1 作為預設值。

可用性

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

Tensor 條件約束

InputTensorOutputTensor 必須具有相同 的 DataType大小

Tensor 支援

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

需求

   
標頭 directml.h