структура DML_MAX_POOLING2_OPERATOR_DESC (directml.h)

Статья
08/23/2023

Вычисляет максимальное значение для элементов в скользящем окне над входным тензором и при необходимости возвращает индексы выбранных максимальных значений.

Синтаксис

struct DML_MAX_POOLING2_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
  const DML_TENSOR_DESC *OutputIndicesTensor;
  UINT                  DimensionCount;
  const UINT            *Strides;
  const UINT            *WindowSize;
  const UINT            *StartPadding;
  const UINT            *EndPadding;
  const UINT            *Dilations;
};

Члены

InputTensor

Тип: const DML_TENSOR_DESC*

Входной тензор размеров{ BatchCount, ChannelCount, Height, Width } , если значение InputTensor.DimensionCount равно 4, а { BatchCount, ChannelCount, Depth, Height, Weight } значение InputTensor.DimensionCount равно 5.

OutputTensor

Тип: const DML_TENSOR_DESC*

Выходной тензор для записи результатов. Размеры выходного тензора можно вычислить следующим образом.

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

OutputIndicesTensor

Тип: _Maybenull_ const DML_TENSOR_DESC*

Необязательный выходной тензор индексов входного тензора InputTensor максимальных значений, созданных и хранящихся в OutputTensor. Эти значения индекса отсчитываются от нуля и рассматривают входной тензор как непрерывный одномерный массив. Если несколько элементов в скользящем окне имеют одинаковое значение, последующие равные значения игнорируются, а индекс указывает на первое обнаруженное значение. И OutputTensor , и OutputIndicesTensor имеют одинаковые размеры тензоров.

DimensionCount

Тип: UINT

Количество пространственных измерений входного тензора InputTensor, которое также соответствует количеству измерений скользящего окна WindowSize. Это значение также определяет размер массивов Strides, StartPadding и EndPadding . Он должен иметь значение 2, если InputTensor имеет 4D, и 3, если это 5D-тензор.

Strides