共用方式為


DML_ELEMENT_WISE_IS_INFINITY_OPERATOR_DESC結構 (directml.h)

根據指定的 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中引進的。

Tensor 條件約束

InputTensorOutputTensor 必須具有相同 的 DimensionCount大小

Tensor 支援

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