DML_ACTIVATION_SHRINK_OPERATOR_DESC 結構 (directml.h)

發行項
02/28/2024

在 InputTensor 中的每個元素上執行壓縮啟用函式，將結果放入 OutputTensor 的對應元素中。

f(x) = x - Bias, if x > Threshold
       x + Bias, if x < -Threshold
       0,        otherwise

這個運算符支援就地執行，這表示輸出張量在系結期間允許 將 InputTensor 別名。

語法

struct DML_ACTIVATION_SHRINK_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
  FLOAT                 Bias;
  FLOAT                 Threshold;
};

成員

InputTensor

類型： const DML_TENSOR_DESC*

要從中讀取的輸入張量。

OutputTensor

類型： const DML_TENSOR_DESC*

要寫入結果的輸出張量。

Bias

類型： FLOAT

偏差的值。這個值的典型預設值為 0.0。

Threshold

類型： FLOAT

臨界值。這個值的典型預設值為 0.5。

備註

可用性

這個運算子是在中 DML_FEATURE_LEVEL_2_0引進的。

Tensor 條件約束

InputTensor 和 OutputTensor 必須具有相同 的 DataType、 DimensionCount 和大小。

Tensor 支援

DML_FEATURE_LEVEL_5_1和更新版本

張	種類	支援的維度計數	支援的資料類型
InputTensor	輸入	1 到 8	FLOAT32、FLOAT16、INT32、INT16、INT8、UINT32、UINT16、UINT8
OutputTensor	輸出	1 到 8	FLOAT32、FLOAT16、INT32、INT16、INT8、UINT32、UINT16、UINT8

DML_FEATURE_LEVEL_3_0和更新版本

張	種類	支援的維度計數	支援的資料類型
InputTensor	輸入	1 到 8	FLOAT32、FLOAT16
OutputTensor	輸出	1 到 8	FLOAT32、FLOAT16

DML_FEATURE_LEVEL_2_0和更新版本

張	種類	支援的維度計數	支援的資料類型
InputTensor	輸入	4 到 5	FLOAT32、FLOAT16
OutputTensor	輸出	4 到 5	FLOAT32、FLOAT16

規格需求

需求	值
最低支援的用戶端	Windows 10 版本 2004 (10.0;組建 19041)
最低支援的伺服器	Windows Server 版本 2004 (10.0;組建 19041)
標頭	directml.h

共用方式為