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_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 |