DML_MAX_POOLING_GRAD_OPERATOR_DESC-Struktur (directml.h)

Artikel
08/23/2023

Berechnet backpropagation gradients for max pooling (siehe DML_MAX_POOLING2_OPERATOR_DESC).

Stellen Sie sich einen 2x2-DML_MAX_POOLING2_OPERATOR_DESC ohne Auffüllung oder Erweiterung und einen Schritt von 1 vor, der folgendes ausführt.

InputTensor             OutputTensor    IndicesTensor
[[1, 2, 3],   MaxPool    [[4, 4],        [[4, 4],
 [2, 4, 2],     -->       [6, 7]]         [7, 8]]
 [5, 6, 7]]

Das größte Element jedes 2x2-Fensters im Eingabe tensor erzeugt ein Element der Ausgabe. Im Folgenden finden Sie ein Beispiel für die Ausgabe von DML_MAX_POOLING_GRAD_OPERATOR_DESC mit ähnlichen Parametern.

InputTensor   InputGradientTensor            OutputGradientTensor
[[1, 2, 3],       [[1, 2],     MaxPoolGrad   [[0, 0, 0],
 [2, 4, 2],        [4, 5]]         -->        [0, 3, 0],
 [5, 6, 7]]                                   [0, 4, 5]]

Tatsächlich verwendet dieser Operator den InputTensor , um den Index des größten Elements aus jedem Fenster zu bestimmen, und verteilt die Werte von InputGradientTensor auf der Grundlage dieser Indizes in den OutputGradientTensor . Wenn sich Indizes überschneiden, werden die Werte summiert. Alle nicht referenzierten Ausgabeelemente werden auf null gesetzt.

Im Fall einer Verknüpfung (bei der mehrere Elemente in einem Fenster den gleichen Maximalwert haben), wird das Element mit dem niedrigsten Index des logischen Elements ausgewählt.

Syntax

struct DML_MAX_POOLING_GRAD_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *InputGradientTensor;
  const DML_TENSOR_DESC *OutputGradientTensor;
  UINT                  DimensionCount;
  const UINT            *Strides;
  const UINT            *WindowSize;
  const UINT            *StartPadding;
  const UINT            *EndPadding;
  const UINT            *Dilations;
};

Member

InputTensor

Typ: const DML_TENSOR_DESC*

Der Tensor des Eingabefeatures. Dies ist in der Regel derselbe Tensor, der wie der InputTensor bereitgestellt wurde, um im Vorwärtsdurchlauf DML_MAX_POOLING2_OPERATOR_DESC .

InputGradientTensor

Typ: const DML_TENSOR_DESC*

Der eingehende Gradienten tensor. Dies wird in der Regel aus der Ausgabe der Backpropagation einer vorangehenden Ebene abgerufen. In der Regel hat dieser Tensor die gleichen Größen wie die Ausgabe des entsprechenden DML_MAX_POOLING2_OPERATOR_DESC im Vorwärtsdurchlauf.

OutputGradientTensor

Typ: const DML_TENSOR_DESC*

Ein Ausgabe tensor, der die umgekehrten Farbverläufe enthält. In der Regel hat dieser Tensor die gleichen Größen wie die Eingabe des entsprechenden DML_MAX_POOLING2_OPERATOR_DESC im Vorwärtsdurchlauf.

DimensionCount

Typ: UINT

Die Anzahl der Elemente in den Arrays Strides, WindowSize, StartPadding, EndPadding und Dilations . Dieser Wert muss der Anzahl räumlicher Dimensionen entsprechen (DimensionCount von InputTensor – 2). Da dieser Operator nur 4D-Tensoren unterstützt, ist der einzige gültige Wert für diesen Parameter 2.

Strides