共用方式為

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 條件約束

InputTensor 和 OutputTensor 必須具有相同的 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