次の方法で共有


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

入力テンソル InputTensor の空間ディメンションの数。これは、スライディング ウィンドウ WindowSize のディメンションの数にも対応します。 この値は、 StridesStartPaddingおよび 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 に設定されている場合、または { Depth, Height, Width } 3 に設定されている場合の スライディング ウィンドウ{ Height, Width }のディメンション。

StartPadding

型: _Field_size_(DimensionCount) const UINT*

入力テンソル InputTensor の各空間ディメンションの先頭に適用されるパディング要素の数。 値は、DimensionCount{ Height, Width } 2 に設定されている場合、または { Depth, Height, Width } 3 に設定されている場合は です。

EndPadding

型: _Field_size_(DimensionCount) const UINT*

入力テンソル InputTensor の各空間ディメンションの末尾に適用されるパディング要素の数。 値は、DimensionCount{ Height, Width } 2 に設定されている場合、または { Depth, Height, Width } 3 に設定されている場合は です。

IncludePadding

種類: BOOL

スライディング ウィンドウ内のすべての要素の平均値を計算するときに、空間エッジの周囲にパディング要素を含めるかどうかを示します。 値が FALSE に設定されている場合、埋め込み要素は平均計算の除数値の一部としてカウントされません。

可用性

この演算子は、 で DML_FEATURE_LEVEL_1_0導入されました。

テンソル制約

InputTensorOutputTensor には、同じ DataTypeDimensionCount が必要です。

Tensor のサポート

テンソル 種類 サポートされているディメンション数 サポートされるデータ型
InputTensor 入力 4 から 5 FLOAT32、FLOAT16
OutputTensor 出力 4 から 5 FLOAT32、FLOAT16

要件

要件
Header directml.h