estructura DML_AVERAGE_POOLING1_OPERATOR_DESC (directml.h)

Promedios de valores en los elementos dentro de la ventana deslizante sobre el tensor de entrada.

Sintaxis

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

Miembros

InputTensor

Tipo: const DML_TENSOR_DESC*

Tensor de entrada de Tamaños{ BatchCount, ChannelCount, Height, Width } para 4D y { BatchCount, ChannelCount, Depth, Height, Weight } para 5D.

OutputTensor

Tipo: const DML_TENSOR_DESC*

Descripción del tensor de salida. Los tamaños del tensor de salida se pueden calcular de la siguiente manera.

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

Número de dimensiones espaciales del tensor de entrada InputTensor, que también corresponde al número de dimensiones de la ventana deslizante WindowSize. Este valor también determina el tamaño de las matrices Strides, StartPadding y EndPadding . Debe establecerse en 2 cuando InputTensor es 4D y 3 cuando es un tensor 5D.

Strides

Tipo: _Field_size_(DimensionCount) const UINT*

Los pasos de las dimensiones de la ventana deslizante de tamaños { Height, Width } cuando DimensionCount se establece en 2 o { Depth, Height, Width } cuando se establece en 3.

WindowSize

Tipo: _Field_size_(DimensionCount) const UINT*

Las dimensiones de la ventana deslizante en cuando { Height, Width } se establece en 2 o { Depth, Height, Width } cuando se establece en 3.

StartPadding

Tipo: _Field_size_(DimensionCount) const UINT*

Número de elementos de relleno que se aplicarán al principio de cada dimensión espacial del tensor de entrada InputTensor. Los valores se encuentran en cuando { Height, Width } se establece en 2 o cuando se establece en { Depth, Height, Width } 3.

EndPadding

Tipo: _Field_size_(DimensionCount) const UINT*

Número de elementos de relleno que se van a aplicar al final de cada dimensión espacial del tensor de entrada InputTensor. Los valores se encuentran en cuando { Height, Width } se establece en 2 o cuando se establece en { Depth, Height, Width } 3.

Dilations

Tipo: _Field_size_(DimensionCount) const UINT*

Los valores de cada dimensión espacial del tensor de entrada InputTensor por el que se selecciona un elemento dentro de la ventana deslizante para cada elemento de ese valor. Los valores se encuentran en cuando { Height, Width } se establece en 2 o cuando se establece en { Depth, Height, Width } 3.

IncludePadding

Tipo: BOOL

Indica si se deben incluir los elementos de relleno alrededor de los bordes espaciales al calcular el valor medio en todos los elementos de la ventana deslizante. Cuando el valor se establece en FALSE, los elementos de relleno no se cuentan como parte del valor divisor del cálculo de promedio.

Observaciones

DML_AVERAGE_POOLING1_OPERATOR_DESC es como DML_AVERAGE_POOLING_OPERATOR_DESC, excepto con una matriz constante adicional Dilations. Cuando Dilations se establece en { 1,1 } para la entrada 4D, o { 1,1,1 } para las características de entrada 5D, DML_AVERAGE_POOLING1_OPERATOR_DESC es equvalente a DML_AVERAGE_POOLING_OPERATOR_DESC.

Disponibilidad

Este operador se introdujo en DML_FEATURE_LEVEL_6_2.

Restricciones tensor

InputTensor y OutputTensor deben tener el mismo DataType y DimensionCount.

Compatibilidad con Tensor

Requisitos