Share via


estructura DML_LP_POOLING_OPERATOR_DESC (directml.h)

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

Sintaxis

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

Miembros

InputTensor

Tipo: const DML_TENSOR_DESC*

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

OutputTensor

Tipo: const DML_TENSOR_DESC*

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 input 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 de las dimensiones de 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*

Matriz que contiene las dimensiones de la ventana deslizante en cuando DimensionCount se establece en { Height, Width }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 van a aplicar al principio de cada dimensión espacial del tensor inputTensor. Los valores se encuentran en cuando DimensionCount se establece en { Height, Width } 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 van a aplicar al final de cada dimensión espacial del tensor inputTensor. Los valores se encuentran en cuando DimensionCount se establece en { Height, Width } 2 o { Depth, Height, Width } cuando se establece en 3.

P

Tipo: UINT

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

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

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

Requisitos

Requisito Valor
Header directml.h