DML_AVERAGE_POOLING1_OPERATOR_DESC structuur (directml.h)

Gemiddelden waarden over de elementen in het schuifvenster boven de invoertenor.

Syntaxis

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;
};

Leden

InputTensor

Type: const DML_TENSOR_DESC*

Een invoertenant van grootten{ BatchCount, ChannelCount, Height, Width } voor 4D en { BatchCount, ChannelCount, Depth, Height, Weight } voor 5D.

OutputTensor

Type: const DML_TENSOR_DESC*

Een beschrijving van de uitvoertensor. De grootten van de uitvoertensor kunnen als volgt worden berekend.

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

Soort: UINT

Het aantal ruimtelijke dimensies van de invoertensor InputTensor, dat ook overeenkomt met het aantal dimensies van het schuifvenster WindowSize. Deze waarde bepaalt ook de grootte van de matrices Strides, StartPadding en EndPadding . Deze moet worden ingesteld op 2 wanneer InputTensor 4D is en 3 wanneer het een 5D-tensor is.

Strides

Type: _Field_size_(DimensionCount) const UINT*

De stappen voor de afmetingen van de schuifvenstergrootten { Height, Width } wanneer de DimensionCount is ingesteld op 2 of { Depth, Height, Width } op 3.

WindowSize

Type: _Field_size_(DimensionCount) const UINT*

De afmetingen van het schuifvenster wanneer { Height, Width }DimensionCount is ingesteld op 2 of { Depth, Height, Width } op 3.

StartPadding

Type: _Field_size_(DimensionCount) const UINT*

Het aantal opvullingselementen dat moet worden toegepast op het begin van elke ruimtelijke dimensie van de invoertensor InputTensor. De waarden bevinden zich in { Height, Width } wanneer DimensionCount is ingesteld op 2 of { Depth, Height, Width } wanneer deze is ingesteld op 3.

EndPadding

Type: _Field_size_(DimensionCount) const UINT*

Het aantal opvullingselementen dat moet worden toegepast op het einde van elke ruimtelijke dimensie van de invoertensor InputTensor. De waarden bevinden zich in { Height, Width } wanneer DimensionCount is ingesteld op 2 of { Depth, Height, Width } wanneer deze is ingesteld op 3.

Dilations

Type: _Field_size_(DimensionCount) const UINT*

De waarden voor elke ruimtelijke dimensie van de invoertensor InputTensor waarmee een element in het schuifvenster wordt geselecteerd voor elk element van die waarde. De waarden bevinden zich in { Height, Width } wanneer DimensionCount is ingesteld op 2 of { Depth, Height, Width } wanneer deze is ingesteld op 3.

IncludePadding

Soort: BOOL

Geeft aan of de opvullingselementen rond de ruimtelijke randen moeten worden opgenomen bij het berekenen van de gemiddelde waarde voor alle elementen in het schuifvenster. Wanneer de waarde is ingesteld op ONWAAR, worden de opvullingselementen niet meegeteld als onderdeel van de delerwaarde van de gemiddeldeberekening.

Opmerkingen

DML_AVERAGE_POOLING1_OPERATOR_DESC lijkt op DML_AVERAGE_POOLING_OPERATOR_DESC, met uitzondering van een extra constante matrixverwijdering. Wanneer dilations is ingesteld op { 1,1 } voor 4D-invoer of { 1,1,1 } voor 5D-invoerfuncties, is DML_AVERAGE_POOLING1_OPERATOR_DESC equvalent tot DML_AVERAGE_POOLING_OPERATOR_DESC.

Beschikbaarheid

Deze operator is geïntroduceerd in DML_FEATURE_LEVEL_6_2.

Tensor-beperkingen

InputTensor en OutputTensor moeten hetzelfde DataType en DimensionCount hebben.

Ondersteuning voor Tensor

Behoeften