DML_AVERAGE_POOLING_OPERATOR_DESC 結構 (directml.h)

在輸入張量上滑動視窗內，跨元素的平均值。

語法

struct DML_AVERAGE_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;
  BOOL                  IncludePadding;
};

成員

InputTensor

類型： const DML_TENSOR_DESC*

4D 和 5D { BatchCount, ChannelCount, Depth, Height, Weight }的大小{ BatchCount, ChannelCount, Height, Width }輸入張數。

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

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

Strides

類型：_Field_size_ (DimensionCount) const UINT*

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

WindowSize

類型：_Field_size_ (DimensionCount) const UINT*

當 DimensionCount 設定為 2 或設定為 3 時，{ Depth, Height, Width }滑動視窗{ Height, Width }的維度。

StartPadding

類型：_Field_size_ (DimensionCount) const UINT*

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

EndPadding

類型：_Field_size_ (DimensionCount) const UINT*

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

IncludePadding

類型： BOOL

指出在計算滑動視窗內所有元素的平均值時，是否要在空間邊緣周圍包含填補元素。 當此值設定為 FALSE 時，填補元素不會計算為平均計算的除數值的一部分。

可用性

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

Tensor 條件約束

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

Tensor 支援

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

規格需求

需求	值
標頭	directml.h