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。
重要
此 API 作为 DirectML 独立可再发行组件包的一部分提供（请参阅 Microsoft.AI.DirectML 版本 1.9 及更高版本。 另请参阅 DirectML 版本历史记录。
语法
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 支持
| Tensor | 种类 | 支持的维度计数 | 支持的数据类型 |
|---|---|---|---|
| InputTensor | 输入 | 1 到 8 | FLOAT32, FLOAT16 |
| OutputTensor | 输出 | 1 到 8 | FLOAT32, FLOAT16 |
要求
| 页眉 | directml.h |