DML_LP_NORMALIZATION_OPERATOR_DESC 結構 (directml.h)

沿著輸入 Tensor 的指定座標軸執行 Lp 正規化函式。

語法

struct DML_LP_NORMALIZATION_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
  UINT                  Axis;
  FLOAT                 Epsilon;
  UINT                  P;
};

成員

InputTensor

類型:const DML_TENSOR_DESC*

包含輸入資料的張量。

OutputTensor

類型:const DML_TENSOR_DESC*

要寫入結果的張量。 這個 Tensor 的大小 應該符合 InputTensor

Axis

類型: UINT

要套用正規化的座標軸。

Epsilon

類型: FLOAT

要用來避免除以零的 epsilon 值。 建議預設值為 0.00001。

P

類型: UINT

正規化的順序 (1 或 2) 。

可用性

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

Tensor 條件約束

InputTensorOutputTensor 必須具有相同 的 DataTypeDimensionCount大小

Tensor 支援

DML_FEATURE_LEVEL_3_1和更新版本

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

DML_FEATURE_LEVEL_1_0和更新版本

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

需求

   
標頭 directml.h