directml.h) (DML_ACTIVATION_PARAMETRIC_SOFTPLUS_OPERATOR_DESC 结构
对 InputTensor 中的每个元素执行参数化软增强激活函数,将结果放入 OutputTensor 的相应元素中。
f(x) = Alpha * ln(1 + exp(Beta * x))
其中 exp (x) 是自然指数函数,ln (x) 是自然对数。
此运算符支持就地执行,这意味着允许输出张量在绑定期间为 InputTensor 别名。
语法
struct DML_ACTIVATION_PARAMETRIC_SOFTPLUS_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 系数。
Beta
类型: FLOAT
beta 系数。
可用性
此运算符是在 中 DML_FEATURE_LEVEL_1_0
引入的。
张量约束
InputTensor 和 OutputTensor 必须具有相同的 DataType、 DimensionCount 和 Size。
张量支持
DML_FEATURE_LEVEL_3_0 及更高版本
张 | 种类 | 支持的维度计数 | 支持的数据类型 |
---|---|---|---|
InputTensor | 输入 | 1 到 8 | FLOAT32、FLOAT16 |
OutputTensor | 输出 | 1 到 8 | FLOAT32、FLOAT16 |
DML_FEATURE_LEVEL_2_0及更高版本
张 | 种类 | 支持的维度计数 | 支持的数据类型 |
---|---|---|---|
InputTensor | 输入 | 4 到 5 | FLOAT32、FLOAT16 |
OutputTensor | 输出 | 4 到 5 | FLOAT32、FLOAT16 |
DML_FEATURE_LEVEL_1_0 及更高版本
张 | 种类 | 支持的维度计数 | 支持的数据类型 |
---|---|---|---|
InputTensor | 输入 | 4 | FLOAT32、FLOAT16 |
OutputTensor | 输出 | 4 | FLOAT32、FLOAT16 |
要求
要求 | 值 |
---|---|
Header | directml.h |