DML_ACTIVATION_THRESHOLDED_RELU_OPERATOR_DESC 结构 (directml.h)
对 InputTensor 中的每个元素执行阈值整流线性单位 (ReLU) 激活函数，并将结果放入 OutputTensor 的相应元素中。
f(x) = x, if x > Alpha
0, otherwise
此运算符支持就地执行，这意味着允许输出张量在绑定期间为 InputTensor 别名。
语法
struct DML_ACTIVATION_THRESHOLDED_RELU_OPERATOR_DESC {
const DML_TENSOR_DESC *InputTensor;
const DML_TENSOR_DESC *OutputTensor;
FLOAT Alpha;
};
成员
InputTensor
类型： const DML_TENSOR_DESC*
要从中读取的输入张量。
OutputTensor
类型： const DML_TENSOR_DESC*
要写入结果的输出张量。
Alpha
类型： FLOAT
函数的阈值。 此值的典型默认值为 1.0。
可用性
此运算符是在 中 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 |
要求
| 标头 | directml.h |