Compartir a través de

estructura DML_MAX_POOLING_OPERATOR_DESC (directml.h)

  • Artículo

Calcula el valor máximo en los elementos de la ventana deslizante sobre el tensor de entrada.

Sintaxis

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

Miembros

InputTensor

Tipo: const DML_TENSOR_DESC*

Un tensor de entrada de tamaños de { BatchCount, ChannelCount, Height, Width } si inputTensor.DimensionCount es 4 y { BatchCount, ChannelCount, Depth, Height, Weight } si inputTensor.DimensionCount es 5.

OutputTensor

Tipo: const DML_TENSOR_DESC*

Tensor de salida en el que escribir los resultados. 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 de

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 Strides, StartPaddingy matrices de EndPadding. Debe establecerse en 2 cuando inputTensor es 4D y 3 cuando es un tensor 5D.

Strides

Tipo: const * UINT

Los pasos de las dimensiones de ventana deslizante de tamaños cuando el DimensionCount de se establece en 2 o cuando se establece en 3.

WindowSize

Tipo: _Field_size_(DimensionCount) const UINT*

Las dimensiones de la ventana deslizante de { Height, Width } cuando dimensionCount 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 { Height, Width } cuando DimensionCount se establece en 2 o { Depth, Height, Width } cuando se establece en 3.

EndPadding

Tipo: _Field_size_(DimensionCount) const UINT*

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

Observaciones

En DML_FEATURE_LEVEL_2_0se introdujo una versión más reciente de este operador, DML_MAX_POOLING1_OPERATOR_DESC.

Disponibilidad

Este operador se introdujo en DML_FEATURE_LEVEL_1_0.

Restricciones tensor

InputTensor y OutputTensor deben tener el mismo DataType y DimensionCount.

Compatibilidad con Tensor

DML_FEATURE_LEVEL_5_0 y versiones posteriores

Tensor Amable Recuentos de dimensiones admitidos Tipos de datos admitidos
InputTensor Entrada De 4 a 5 FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8
OutputTensor Salida De 4 a 5 FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8

DML_FEATURE_LEVEL_3_0 y versiones posteriores

Tensor Amable Recuentos de dimensiones admitidos Tipos de datos admitidos
InputTensor Entrada De 4 a 5 FLOAT32, FLOAT16, INT8, UINT8
OutputTensor Salida De 4 a 5 FLOAT32, FLOAT16, INT8, UINT8

DML_FEATURE_LEVEL_1_0 y versiones posteriores

Tensor Amable Recuentos de dimensiones admitidos Tipos de datos admitidos
InputTensor Entrada De 4 a 5 FLOAT32, FLOAT16
OutputTensor Salida De 4 a 5 FLOAT32, FLOAT16

Requisitos

Requisito Valor
encabezado de directml.h