DML_AVERAGE_POOLING1_OPERATOR_DESC Struktur (directml.h)

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

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

Elemente

InputTensor

Typ: const DML_TENSOR_DESC*

Ein Eingabe-Tensor von 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-Tensors. 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 Anzahl der räumlichen Dimensionen des Eingabe-Tensor InputTensor, der auch der Anzahl 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 { Height, Width } für das gleitende Fenster, wenn der DimensionCount auf 2 festgelegt ist, oder { Depth, Height, Width } wenn sie auf 3 festgelegt ist.

WindowSize

Typ: _Field_size_(DimensionCount) const UINT*

Die Abmessungen des Gleitfensters, wenn { Height, Width }DimensionCount auf 2 festgelegt ist, oder { Depth, Height, Width } wenn sie 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, { 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, { 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 Gleitfensters für jedes Element dieses Werts ausgewählt wird. Die Werte befinden sich, { 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.

Bemerkungen

DML_AVERAGE_POOLING1_OPERATOR_DESC ist wie DML_AVERAGE_POOLING_OPERATOR_DESC, mit Ausnahme einer zusätzlichen Konstantenarray-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 quvalent für DML_AVERAGE_POOLING_OPERATOR_DESC.

Verfügbarkeit

Dieser Operator wurde in DML_FEATURE_LEVEL_6_2 eingeführt.

Tensoreinschränkungen

"InputTensor " und " OutputTensor " müssen denselben Datentyp und "DimensionCount" aufweisen.

Tensorunterstützung

Anforderungen