DML_ACTIVATION_GELU_OPERATOR_DESC 结构 (directml.h)

对 InputTensor 中的每个元素执行高斯误差线性单元 (GELU) 激活函数，将结果放入 OutputTensor 的相应元素中。

f(x) = 0.5 * x * (1.0 + erf(x / sqrt(2)))

其中 erf(x) 是 DML_ELEMENT_WISE_ERF_OPERATOR_DESC。

语法

struct DML_ACTIVATION_GELU_OPERATOR_DESC
{
    const DML_TENSOR_DESC* InputTensor;
    const DML_TENSOR_DESC* OutputTensor;
};

成员

InputTensor

类型：const DML_TENSOR_DESC*

要从中读取数据的输入张量。

OutputTensor

类型：const DML_TENSOR_DESC*

要向其写入结果的输出张量。

可用性

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

Tensor 约束

InputTensor 和 OutputTensor 必须具有相同的 DataType、DimensionCount 和 Sizes。

Tensor 支持

要求