共用方式為


DML_ACTIVATION_RELU_OPERATOR_DESC 結構 (directml.h)

InputTensor 的每個元素上執行已修正的線性單位 (ReLU) 啟用函式,並將結果放入 OutputTensor 的對應元素中。

f(x) = max(0, x)

其中 max (a,b) 傳回兩個值 a、b 的較大值。

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

語法

struct DML_ACTIVATION_RELU_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
};

成員

InputTensor

類型: const DML_TENSOR_DESC*

要從中讀取的輸入張量。

OutputTensor

類型: const DML_TENSOR_DESC*

要寫入結果的輸出張量。

可用性

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

Tensor 條件約束

InputTensorOutputTensor 必須具有相同 的 DataTypeDimensionCount大小

Tensor 支援

DML_FEATURE_LEVEL_5_1和更新版本

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

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

DML_FEATURE_LEVEL_1_0和更新版本

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

規格需求

需求
標頭 directml.h