Freigeben über

DML_MAX_POOLING_OPERATOR_DESC Struktur (directml.h)

  • Artikel

Berechnet den Maximalwert über die Elemente innerhalb des Gleitfensters über den Eingabe-Tensor.

Syntax

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

Angehörige

InputTensor

Typ: DML_TENSOR_DESC*

Ein Eingabe-Tensor von Größen{ BatchCount, ChannelCount, Height, Width }, wenn InputTensor.DimensionCount 4 ist, und { BatchCount, ChannelCount, Depth, Height, Weight }, wenn InputTensor.DimensionCount 5 ist.

OutputTensor

Typ: DML_TENSOR_DESC*

Ein Ausgabe-Tensor, in den die Ergebnisse geschrieben werden sollen. 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-Tensors InputTensor, die auch der Anzahl der Dimensionen des gleitenden Fensters WindowSizeentspricht. Dieser Wert bestimmt auch die Größe der Strides, StartPaddingund EndPadding Arrays. Es sollte auf 2 festgelegt werden, wenn InputTensor- 4D ist, und 3, wenn es sich um einen 5D-Tensor handelt.

Strides

Typ: UINT-*

Die Schritte für die Größenabmessungen des Gleitfensters { Height, Width }, wenn die DimensionCount- auf 2 festgelegt ist, oder { Depth, Height, Width }, wenn sie auf 3 festgelegt ist.

WindowSize

Typ: _Field_size_(DimensionCount) UINT-*

Die Abmessungen des Gleitfensters in { Height, Width }, wenn DimensionCount- auf 2 festgelegt ist, oder { Depth, Height, Width }, wenn sie auf 3 festgelegt ist.

StartPadding

Typ: _Field_size_(DimensionCount) UINT-*

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

EndPadding

Typ: _Field_size_(DimensionCount) UINT-*

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

Bemerkungen

Eine neuere Version dieses Operators, DML_MAX_POOLING1_OPERATOR_DESC, wurde in DML_FEATURE_LEVEL_2_0eingeführt.

Verfügbarkeit

Dieser Operator wurde in DML_FEATURE_LEVEL_1_0eingeführt.

Tensoreinschränkungen

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

Tensorunterstützung

DML_FEATURE_LEVEL_5_0 und höher

Tensor Art Unterstützte Dimensionsanzahl Unterstützte Datentypen
InputTensor Eingabe 4 bis 5 FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8
OutputTensor Ausgabe 4 bis 5 FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8

DML_FEATURE_LEVEL_3_0 und höher

Tensor Art Unterstützte Dimensionsanzahl Unterstützte Datentypen
InputTensor Eingabe 4 bis 5 FLOAT32, FLOAT16, INT8, UINT8
OutputTensor Ausgabe 4 bis 5 FLOAT32, FLOAT16, INT8, UINT8

DML_FEATURE_LEVEL_1_0 und höher

Tensor Art Unterstützte Dimensionsanzahl Unterstützte Datentypen
InputTensor Eingabe 4 bis 5 FLOAT32, FLOAT16
OutputTensor Ausgabe 4 bis 5 FLOAT32, FLOAT16

Anforderungen

Anforderung Wert
Header- directml.h