DML_AVERAGE_POOLING1_OPERATOR_DESC構造体 (directml.h)

入力テンソルのスライディング ウィンドウ内の要素全体の値を平均化します。

構文

struct DML_AVERAGE_POOLING1_OPERATOR_DESC
{
    const DML_TENSOR_DESC* InputTensor;
    const DML_TENSOR_DESC* OutputTensor;
    UINT DimensionCount;
    _Field_size_(DimensionCount) const UINT* Strides;
    _Field_size_(DimensionCount) const UINT* WindowSize;
    _Field_size_(DimensionCount) const UINT* StartPadding;
    _Field_size_(DimensionCount) const UINT* EndPadding;
    _Field_size_(DimensionCount) const UINT* Dilations;
    BOOL IncludePadding;
};

メンバー

InputTensor

型: const DML_TENSOR_DESC*

サイズの入力テンソル{ BatchCount, ChannelCount, Height, Width } 4D の場合は { BatchCount, ChannelCount, Depth, Height, Weight }、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 のディメンションの数にも対応します。 この値は、 Strides、 StartPadding、 EndPadding 配列のサイズも決定します。 InputTensor が 4D の場合は 2、5D テンソルの場合は 3 に設定する必要があります。

Strides

型: _Field_size_(DimensionCount) const UINT*

{ Height, Width } が 2 に設定されている場合にサイズのスライディング ウィンドウディメンションのストライド、または 3 に設定すると{ Depth, Height, Width }。

WindowSize

型: _Field_size_(DimensionCount) const UINT*

{ Height, Width } が 2 に設定されている場合はのスライディング ウィンドウの寸法。3 に設定すると{ Depth, Height, Width }。

StartPadding

型: _Field_size_(DimensionCount) const UINT*

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

EndPadding

型: _Field_size_(DimensionCount) const UINT*

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

Dilations

型: _Field_size_(DimensionCount) const UINT*

スライディング ウィンドウ内の要素がその値のすべての要素に対して選択される 、入力テンソル InputTensor の各空間ディメンションの値。 { Height, Width } が 2 に設定されている場合は値が、3 に設定すると{ Depth, Height, Width }されます。

IncludePadding

型: BOOL

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

注釈

DML_AVERAGE_POOLING1_OPERATOR_DESC は、追加の定数配列拡張を除いて、 DML_AVERAGE_POOLING_OPERATOR_DESCのようなもの です。 4D 入力の場合は Dilations が { 1,1 } に、5D 入力機能の場合は { 1,1,1 } に設定されている 場合、DML_AVERAGE_POOLING1_OPERATOR_DESC は DML_AVERAGE_POOLING_OPERATOR_DESC に対して equvalent です。

可用性

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

Tensor 制約

InputTensor と OutputTensor には、同じ DataType と DimensionCount が必要です。

Tensor のサポート

要求事項