通过

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。

  • 如果 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