通过

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，并且还必须匹配输入张量通道维度的大小。

Bias

类型： const FLOAT*

包含输入张量每个维度的偏差项的 FLOAT 值数组。

可用性

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

张量约束

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

张量支持

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

要求

要求 值
Header directml.h