DML_ACTIVATION_SIGMOID_OPERATOR_DESC 结构 (directml.h)
对 InputTensor 中的每个元素执行 sigmoid 函数，将结果放入 OutputTensor 的相应元素中。
f(x) = 1 / (1 + exp(-x))
其中 exp (x) 是自然指数函数。
此运算符支持就地执行，这意味着允许输出张量在绑定期间将 InputTensor 别名。
语法
struct DML_ACTIVATION_SIGMOID_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_1_0。
张量约束
InputTensor 和 OutputTensor 必须具有相同的 DataType、 DimensionCount 和 Size。
Tensor 支持
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 |