Compartir a través de

Estructura DML_LP_POOLING1_OPERATOR_DESC (directml.h)

  • Artículo

Calcula el valor normalizado de LP en los elementos de la ventana deslizante sobre el tensor de entrada.

Importante

Esta API está disponible como parte del paquete redistribuible independiente DirectML (consulte la versión 1.13 y versiones posteriores de Microsoft.AI.DirectML). Consulte también el historial de versiones de DirectML.

Sintaxis

struct DML_LP_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;
    UINT P;
};

Miembros

InputTensor

Tipo: const DML_TENSOR_DESC*

Un tensor de entrada de Sizes{ 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 en el que se va a escribir. 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*

Matriz que contiene los pasos para las dimensiones de la ventana deslizante, de tamaño { Height, Width } cuando DimensionCount se establece en 2 o { Depth, Height, Width } cuando se establece en 3.

WindowSize

Tipo: _Field_size_(DimensionCount) const UINT*

Matriz que contiene las dimensiones de la ventana deslizante, en { Height, Width } cuando DimensionCount se establece en 2 o { Depth, Height, Width } cuando se establece en 3.

StartPadding

Tipo: _Field_size_(DimensionCount) const UINT*

Matriz que contiene el 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*

Matriz que contiene el 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.

Dilations

Tipo: _Field_size_(DimensionCount) const UINT*

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

P

Tipo: UINT

Valor de la variable P en la función de normalización LP Y = (X1^P + X2^P + ... + Xn^P) ^ (1/P), donde el intervalo de X1 a Xn representa cada uno de los valores de la ventana deslizante. En casos de uso comunes, este valor se establece en 1 o 2 y representa la normalización L1 o la L2 respectivamente.

Comentarios

DML_LP_POOLING1_OPERATOR_DESC es como DML_LP_POOLING_OPERATOR_DESC, con la excepción de 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 la entrada 5D, DML_LP_POOLING1_OPERATOR_DESC es equivalente a DML_LP_POOLING_OPERATOR_DESC.

Disponibilidad

Este operador se introdujo en DML_FEATURE_LEVEL_6_2.

Restricciones de tensor

InputTensor y OutputTensor deben tener los mismos DataType y DimensionCount.

Compatibilidad del tensor

Tensor Clase 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

   
Encabezado directml.h