DML_ELEMENT_WISE_IS_NAN_OPERATOR_DESC 结构 (directml.h)
对于输入张量的每个元素,如果输入是 IEEE-754) 定义的 NaN (,则返回 1;否则返回 0。 结果将放入输出张量的相应元素中。
语法
struct DML_ELEMENT_WISE_IS_NAN_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_2_0
。
张量约束
InputTensor 和 OutputTensor 必须具有相同的 DimensionCount 和 Size。
Tensor 支持
DML_FEATURE_LEVEL_3_0及更高版本
张 | 种类 | 支持的维度计数 | 支持的数据类型 |
---|---|---|---|
InputTensor | 输入 | 1 到 8 | FLOAT32、FLOAT16 |
OutputTensor | 输出 | 1 到 8 | UINT32、UINT8 |
DML_FEATURE_LEVEL_2_0及更高版本
张 | 种类 | 支持的维度计数 | 支持的数据类型 |
---|---|---|---|
InputTensor | 输入 | 4 | FLOAT32、FLOAT16 |
OutputTensor | 输出 | 4 | UINT32、UINT8 |
要求
要求 | 值 |
---|---|
最低受支持的客户端 | Windows 10版本 2004 (10.0;内部版本 19041) |
最低受支持的服务器 | Windows Server 版本 2004 (10.0;内部版本 19041) |
标头 | directml.h |