通过

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

要对每个维度求和的元素数：Width、Height 和通道 (（可选）（如果设置了 CrossChannel) ）。 此值必须至少为 1。

Alpha

类型： FLOAT

缩放参数的值。 建议将值 0.0001 作为默认值。

Beta

类型： FLOAT

指数值。 建议将值 0.75 作为默认值。

Bias

类型： FLOAT

偏差的值。 建议将值 1 作为默认值。

可用性

此运算符是在 中 DML_FEATURE_LEVEL_1_0引入的。

张量约束

InputTensor 和 OutputTensor 必须具有相同的 数据类型 和 大小。

张量支持

张 种类 支持的维度计数 支持的数据类型
InputTensor 输入 4 FLOAT32、FLOAT16
OutputTensor 输出 4 FLOAT32、FLOAT16

要求

   
标头 directml.h