Freigeben über


DML_AVERAGE_POOLING1_OPERATOR_DESC-Struktur (directml.h)

Durchschnittliche Werte für die Elemente innerhalb des gleitenden Fensters über den Eingabe-Tensor.

Wichtig

Diese API ist als Teil des eigenständigen weiterverteilbare Pakets DirectML verfügbar (siehe Microsoft.AI.DirectML , Version 1.13 und höher). Siehe auch DirectML-Versionsverlauf.

Syntax

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

Member

InputTensor

Typ: const DML_TENSOR_DESC*

Ein Eingabe-Tensor der Größen{ BatchCount, ChannelCount, Height, Width } für 4D und { BatchCount, ChannelCount, Depth, Height, Weight } für 5D.

OutputTensor

Typ: const DML_TENSOR_DESC*

Eine Beschreibung des Ausgabe-Tensor. Die Größen des Ausgabe-Tensors können wie folgt berechnet werden.

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

Typ: UINT

Die Zahl der räumlichen Dimensionen des Eingabe-Tensor InputTensor, der auch der Zahl der Dimensionen des gleitenden Fensters WindowSize entspricht. Dieser Wert bestimmt auch die Größe der Arrays Strides, StartPadding und EndPadding. Er sollte auf 2 festgelegt werden, wenn InputTensor 4D ist, und 3, wenn es sich um einen 5D-Tensor handelt.

Strides

Typ: _Field_size_(DimensionCount) const UINT*

Die Abstände für die Größenabmessungen für das gleitende Fenster, { Height, Width }wenn der DimensionCount auf 2 festgelegt ist, oder { Depth, Height, Width } wenn er auf 3 festgelegt ist.

WindowSize

Typ: _Field_size_(DimensionCount) const UINT*

Die Abmessungen für das gleitende Fenster, { Height, Width }wenn der DimensionCount auf 2 festgelegt ist, oder { Depth, Height, Width } wenn er auf 3 festgelegt ist.

StartPadding

Typ: _Field_size_(DimensionCount) const UINT*

Die Anzahl der Abstandselemente, die am Anfang jeder räumlichen Dimension des Eingabe-Tensor InputTensor angewendet werden sollen. Die Werte befinden sich in { Height, Width } wenn DimensionCount auf 2 festgelegt ist oder { Depth, Height, Width } wenn sie auf 3 festgelegt ist.

EndPadding

Typ: _Field_size_(DimensionCount) const UINT*

Die Anzahl der Abstandselemente, die am Ende jeder räumlichen Dimension des Eingabe-Tensor InputTensor angewendet werden sollen. Die Werte befinden sich in { Height, Width } wenn DimensionCount auf 2 festgelegt ist oder { Depth, Height, Width } wenn sie auf 3 festgelegt ist.

Dilations

Typ: _Field_size_(DimensionCount) const UINT*

Die Werte für jede räumliche Dimension des Eingabe-Tensor InputTensor, mit dem ein Element innerhalb des gleitenden Fensters für jedes Element dieses Werts ausgewählt wird. Die Werte befinden sich in { Height, Width } wenn DimensionCount auf 2 festgelegt ist oder { Depth, Height, Width } wenn sie auf 3 festgelegt ist.

IncludePadding

Typ: BOOL

Gibt an, ob die Abstandselemente um die räumlichen Kanten beim Berechnen des Mittelwerts für alle Elemente innerhalb des gleitenden Fensters eingeschlossen werden sollen. Wenn der Wert auf FALSE festgelegt ist, werden die Abstandselemente nicht als Teil des Divisorwerts der durchschnittlichen Berechnung gezählt.

Hinweise

DML_AVERAGE_POOLING1_OPERATOR_DESC ist wie DML_AVERAGE_POOLING_OPERATOR_DESC, mit Ausnahme einer zusätzlichen Konstanten-Array-Dilation. Wenn Dilationen für 4D-Eingaben auf { 1,1 } oder { 1,1,1 } für 5D-Eingabefeatures festgelegt sind, ist DML_AVERAGE_POOLING1_OPERATOR_DESC äquivalent für DML_AVERAGE_POOLING_OPERATOR_DESC.

Verfügbarkeit

Dieser Operator wurde in DML_FEATURE_LEVEL_6_2 eingeführt.

Tensor-Einschränkungen

InputTensor und OutputTensor müssen denselben DataType und DimensionCount aufweisen.

Tensor-Unterstützung

Tensor Variante Unterstützte Dimensionsanzahl Unterstützte Datentypen
InputTensor Eingabe 4 bis 5 FLOAT32, FLOAT16
OutputTensor Output 4 bis 5 FLOAT32, FLOAT16

Anforderungen

   
Übergeordnet directml.h