DML_AVERAGE_POOLING_OPERATOR_DESC struttura (directml.h)

Media dei valori tra gli elementi all'interno della finestra scorrevole sul tensore di input.

Sintassi

struct DML_AVERAGE_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;
  BOOL                  IncludePadding;
};

Members

InputTensor

Tipo: const DML_TENSOR_DESC*

Tensore di input di Dimensioni{ BatchCount, ChannelCount, Height, Width } per 4D e { BatchCount, ChannelCount, Depth, Height, Weight } per 5D.

OutputTensor

Tipo: const DML_TENSOR_DESC*

Descrizione del tensore di output. Le dimensioni del tensore di output possono essere calcolate come indicato di seguito.

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

Tipo: UINT

Numero di dimensioni spaziali del tensore di input InputTensor, che corrisponde anche al numero di dimensioni della finestra scorrevole WindowSize. Questo valore determina anche le dimensioni delle matrici Strides, StartPadding e EndPadding. Deve essere impostato su 2 quando InputTensor è 4D e 3 quando è un tensore 5D.

Strides

Tipo: _Field_size_(DimensionCount) const UINT*

Lo stride per le dimensioni della finestra scorrevole { Height, Width } delle dimensioni quando DimensionCount è impostato su 2 o { Depth, Height, Width } quando impostato su 3.

WindowSize

Tipo: _Field_size_(DimensionCount) const UINT*

Dimensioni della finestra scorrevole in { Height, Width } quando DimensionCount è impostato su 2 o { Depth, Height, Width } se impostato su 3.

StartPadding

Tipo: _Field_size_(DimensionCount) const UINT*

Numero di elementi di riempimento da applicare all'inizio di ogni dimensione spaziale del tensore di input InputTensor. I valori si trovano quando { Height, Width }DimensionCount è impostato su 2 o { Depth, Height, Width } quando è impostato su 3.

EndPadding

Tipo: _Field_size_(DimensionCount) const UINT*

Numero di elementi di riempimento da applicare alla fine di ogni dimensione spaziale del tensore inputTensor. I valori si trovano quando { Height, Width }DimensionCount è impostato su 2 o { Depth, Height, Width } quando è impostato su 3.

IncludePadding

Tipo: BOOL

Indica se includere gli elementi di riempimento intorno ai bordi spaziali quando si calcola il valore medio in tutti gli elementi all'interno della finestra scorrevole. Quando il valore è impostato su FALSE, gli elementi di riempimento non vengono conteggiati come parte del valore divisore del calcolo della media.

Disponibilità

Questo operatore è stato introdotto in DML_FEATURE_LEVEL_1_0.

Vincoli tensor

InputTensor e OutputTensor devono avere lo stesso Oggetto DataType e DimensionCount.

Supporto di Tensor

Tensore	Tipo	Conteggi delle dimensioni supportate	Tipi di dati supportati
InputTensor	Input	da 4 a 5	FLOAT32, FLOAT16
OutputTensor	Output	da 4 a 5	FLOAT32, FLOAT16

Requisiti

Requisito	Valore
Intestazione	directml.h