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 條件約束
InputTensor 和 OutputTensor 必須具有相同 的 DataType 和 大小。
Tensor 支援
張 | 種類 | 支援的維度計數 | 支援的資料類型 |
---|---|---|---|
InputTensor | 輸入 | 4 | FLOAT32、FLOAT16 |
OutputTensor | 輸出 | 4 | FLOAT32、FLOAT16 |
需求
標頭 | directml.h |