estrutura DML_AVERAGE_POOLING_OPERATOR_DESC (directml.h)

Médias de valores entre os elementos dentro da janela deslizante sobre o tensor de entrada.

Sintaxe

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

Membros

InputTensor

Tipo: const DML_TENSOR_DESC*

Um tensor de entrada de Tamanhos{ BatchCount, ChannelCount, Height, Width } para 4D e { BatchCount, ChannelCount, Depth, Height, Weight } para 5D.

OutputTensor

Tipo: const DML_TENSOR_DESC*

Uma descrição do tensor de saída. Os tamanhos do tensor de saída podem ser calculados da seguinte maneira.

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

O número de dimensões espaciais do tensor de entrada InputTensor, que também corresponde ao número de dimensões da janela deslizante WindowSize. Esse valor também determina o tamanho das matrizes Strides, StartPadding e EndPadding . Ele deve ser definido como 2 quando InputTensor for 4D e 3 quando for um tensor 5D.

Strides

Tipo: _Field_size_(DimensionCount) const UINT*

Os passos para as dimensões de janela deslizantes de tamanhos { Height, Width } quando o DimensionCount é definido como 2 ou { Depth, Height, Width } quando definido como 3.

WindowSize

Tipo: _Field_size_(DimensionCount) const UINT*

As dimensões da janela deslizante em { Height, Width } quando DimensionCount é definido como 2 ou { Depth, Height, Width } quando definido como 3.

StartPadding

Tipo: _Field_size_(DimensionCount) const UINT*

O número de elementos de preenchimento a serem aplicados ao início de cada dimensão espacial do tensor de entrada InputTensor. Os valores estão em { Height, Width } quando DimensionCount é definido como 2 ou { Depth, Height, Width } quando definido como 3.

EndPadding

Tipo: _Field_size_(DimensionCount) const UINT*

O número de elementos de preenchimento a serem aplicados ao final de cada dimensão espacial do tensor de entrada InputTensor. Os valores estão em { Height, Width } quando DimensionCount é definido como 2 ou { Depth, Height, Width } quando definido como 3.

IncludePadding

Tipo: BOOL

Indica se os elementos de preenchimento devem ser incluídos ao redor das bordas espaciais ao calcular o valor médio em todos os elementos dentro da janela deslizante. Quando o valor é definido como FALSE, os elementos de preenchimento não são contados como parte do valor divisor do cálculo de média.

Disponibilidade

Esse operador foi introduzido em DML_FEATURE_LEVEL_1_0.

Restrições do Tensor

InputTensor e OutputTensor devem ter os mesmos DataType e DimensionCount.

Suporte ao Tensor

Tensor	Tipo	Contagens de dimensões com suporte	Tipos de dados com suporte
InputTensor	Entrada	4 a 5	FLOAT32, FLOAT16
OutputTensor	Saída	4 a 5	FLOAT32, FLOAT16

Requisitos

Requisito	Valor
Cabeçalho	directml.h