通过

DML_ACTIVATION_RELU_GRAD_OPERATOR_DESC 结构 (directml.h)

  • 项目

(ReLU) 计算修正线性单位的反向传播梯度。 此运算符执行以下元素计算。

X = InputTensor
dY = InputGradientTensor

OutputGradientTensor = (X > 0 ? dY : 0)

DML_ACTIVATION_RELU_OPERATOR_DESC相应的正向传递运算符。

语法

struct DML_ACTIVATION_RELU_GRAD_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *InputGradientTensor;
  const DML_TENSOR_DESC *OutputGradientTensor;
};

成员

InputTensor

类型： const DML_TENSOR_DESC*

输入 (特征) 张量。 这通常与前向传递期间提供的输入相同， (请参阅 DML_ACTIVATION_RELU_OPERATOR_DESC) 。

InputGradientTensor

类型： const DML_TENSOR_DESC*

传入的渐变张量。 这通常是从上一层的反向传播的输出中获取的。 此张量的大小和数据类型必须与 InputTensor 的大小和数据类型完全匹配。

OutputGradientTensor

要求

要求 值
最低受支持的客户端 Windows 10内部版本 20348
最低受支持的服务器 Windows 10内部版本 20348
标头 directml.h

另请参阅

DML_ACTIVATION_RELU_OPERATOR_DESC