DML_LP_POOLING_OPERATOR_DESC結構 （directml.h）

計算輸入張量上滑動視窗內元素的 Lp 正規化值。

語法

struct DML_LP_POOLING_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
  UINT                  DimensionCount;
  const UINT            *Strides;
  const UINT            *WindowSize;
  const UINT            *StartPadding;
  const UINT            *EndPadding;
  UINT                  P;
};

成員

InputTensor

類型：const DML_TENSOR_DESC*

具有 4D 大小的輸入張量，以及 5D 的 。

OutputTensor

類型：const DML_TENSOR_DESC*

要寫入的輸出張量。 輸出張量 大小可以計算如下。

OutputTensor->Sizes[0] = InputTensor->Sizes[0];
OutputTensor->Sizes[1] = InputTensor->Sizes[1];

for (UINT i = 0; i < DimensionCount; ++i) {
  UINT PaddedSize = InputTensor->Sizes[i + 2] + StartPadding[i] + EndPadding[i];
  OutputTensor->Sizes[i + 2] = (PaddedSize - WindowSizes[i]) / Strides[i] + 1;
}

DimensionCount

類型：UINT

輸入張量 InputTensor的空間維度數目，這也對應於 WindowSize WindowSize的滑動視窗維度數目。 這個值也會決定 Strides、StartPadding和 EndPadding 數位列的大小。 當 inputTensor 為 4D 時，它應該設定為 2，而當它是 5D 張量時，則設定為 3。

Strides

類型：_Field_size_（DimensionCount） const UINT*

數位列，包含當 DimensionCount 設定為 2，{ Depth, Height, Width } 或設定為 3 時，{ Height, Width } 滑動視窗維度的步幅。

WindowSize

類型：_Field_size_（DimensionCount） const UINT*

當 dimensionCount 設定為 2 時，陣列，包含 中滑動視窗的維度，或設定為 3 時 。

StartPadding

類型：_Field_size_（DimensionCount） const UINT*

數位列，包含要套用至輸入張量之每個空間維度開頭的填補元素數目，InputTensor。 當 dimensionCount 設定為 2，或設定為 3 時 { Depth, Height, Width }，這些值會處於 { Height, Width }。

EndPadding

類型：_Field_size_（DimensionCount） const UINT*

數位列，包含要套用至輸入張量之每個空間維度結尾的填補元素數目，InputTensor。 當 dimensionCount 設定為 2，或設定為 3 時 { Depth, Height, Width }，這些值會處於 { Height, Width }。

P

類型：UINT

Lp 正規化函式中的 P 變數值 Y = (X1^P + X2^P + ... + Xn^P) ^ (1/P)，其中 X1Xn 代表滑動視窗中每個值。 在常見的使用案例中，此值會分別設定為 1 或 2，分別代表 L1 或 L2 正規化。

可用性

此運算子是在 DML_FEATURE_LEVEL_1_0中引進的。

Tensor 條件約束

InputTensor 和 OutputTensor 必須具有相同 的 DataType，且 DimensionCount。

Tensor 支援

張肌	類	支援的維度計數	支援的數據類型
InputTensor	輸入	4 到 5	FLOAT32，FLOAT16
OutputTensor	輸出	4 到 5	FLOAT32，FLOAT16

要求

要求	價值
標頭	directml.h