directml.h) (DML_ELEMENT_WISE_IS_INFINITY_OPERATOR_DESC 结构
根据给定的 InfinityMode 检查 InputTensor 的每个元素是否为 IEEE-754 -inf 和/或 inf,并将结果 (1 表示 true,0 表示 false) 放入 OutputTensor 的相应元素中。
f(x) = isinf(x) && (
(x > 0 && InfinityMode == DML_IS_INFINITY_MODE_POSITIVE) ||
(x < 0 && InfinityMode == DML_IS_INFINITY_MODE_NEGATIVE) ||
InfinityMode == DML_IS_INFINITY_MODE_EITHER)
语法
struct DML_ELEMENT_WISE_IS_INFINITY_OPERATOR_DESC {
const DML_TENSOR_DESC *InputTensor;
const DML_TENSOR_DESC *OutputTensor;
DML_IS_INFINITY_MODE InfinityMode;
};
成员
InputTensor
类型: const DML_TENSOR_DESC*
要从中读取的输入张量。
OutputTensor
类型: const DML_TENSOR_DESC*
要写入结果的输出张量。
InfinityMode
确定要为其检查无穷大符号的DML_IS_INFINITY_MODE。
- 如果 DML_IS_INFINITY_MODE_EITHER,则如果元素为 -inf 或 inf,则返回 1,否则返回 0。
- 如果 DML_IS_INFINITY_MODE_POSITIVE,则如果元素为 inf,则返回 1,否则返回 0。
- 如果 DML_IS_INFINITY_MODE_NEGATIVE',则如果元素为 -inf,则返回 1,否则返回 0。
备注
可用性
此运算符是在 中 DML_FEATURE_LEVEL_2_1
引入的。
张量约束
InputTensor 和 OutputTensor 必须具有相同的 DimensionCount 和 Size。
张量支持
DML_FEATURE_LEVEL_3_0 及更高版本
张 | 种类 | 支持的维度计数 | 支持的数据类型 |
---|---|---|---|
InputTensor | 输入 | 1 到 8 | FLOAT32、FLOAT16 |
OutputTensor | 输出 | 1 到 8 | UINT8 |
DML_FEATURE_LEVEL_2_1 及更高版本
张 | 种类 | 支持的维度计数 | 支持的数据类型 |
---|---|---|---|
InputTensor | 输入 | 4 | FLOAT32、FLOAT16 |
OutputTensor | 输出 | 4 | UINT8 |
要求
最低受支持的客户端 | Windows 10,版本 2004 (10.0;内部版本 19041) |
最低受支持的服务器 | Windows Server 版本 2004 (10.0;内部版本 19041) |
标头 | directml.h |