DML_ELEMENT_WISE_IS_INFINITY_OPERATOR_DESC结构（directml.h）

根据给定的 InfinityMode，检查 InputTensor 的每个元素，并将结果 （1 for true， 0 for 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中引入的。

Tensor 约束

InputTensor 和 OutputTensor 必须具有相同 的 DimensionCount 和 大小。

Tensor 支持

DML_FEATURE_LEVEL_3_0及更高版本

DML_FEATURE_LEVEL_2_1及更高版本

要求