通过

DML_ACTIVATION_HARD_SWISH_OPERATOR_DESC 结构 (directml.h)

  • 项目

对 InputTensor 中的每个元素执行 hard swish 激活函数，将结果放入 OutputTensor 的相应元素中。

f(x) = x * HardSigmoid(x, Alpha, Beta)

此运算符支持就地执行，这意味着在绑定期间将允许输出张量别名设为 InputTensor。

重要

此 API 作为 DirectML 独立可再发行组件包的一部分提供（请参阅 Microsoft.AI.DirectML 版本 1.13 及更高版本。 另请参阅 DirectML 版本历史记录。

语法

struct DML_ACTIVATION_HARD_SWISH_OPERATOR_DESC
{
    const DML_TENSOR_DESC* InputTensor;
    const DML_TENSOR_DESC* OutputTensor;
    FLOAT Alpha;
    FLOAT Beta;
};

成员

InputTensor

类型：const DML_TENSOR_DESC*

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

OutputTensor

类型：const DML_TENSOR_DESC*

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

Alpha

类型：FLOAT

alpha 系数。 该值的典型默认值为 0.2。

Beta

类型：FLOAT

beta 系数。

可用性

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

Tensor 约束

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

Tensor 支持

Tensor 种类 支持的维度计数 支持的数据类型
InputTensor 输入 1 到 8 FLOAT32, FLOAT16
OutputTensor 输出 1 到 8 FLOAT32, FLOAT16

要求

   
页眉 directml.h