DML_MAX_POOLING_OPERATOR_DESC结构（directml.h）

计算输入张量上滑动窗口中元素的最大值。

语法

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

成员

InputTensor

类型：const DML_TENSOR_DESC*

如果 InputTensor.DimensionCount 为 4，则 大小的输入张量{ BatchCount, ChannelCount, Height, Width }，如果 InputTensor.DimensionCount 为 5，则 { BatchCount, ChannelCount, Depth, Height, Weight }。

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

DimensionCount

类型：UINT

输入张量 InputTensor的空间维度数，这也对应于滑动窗口 WindowSize的维度数。 此值还确定 Strides、StartPadding和 EndPadding 数组的大小。 当 InputTensor 为 4D 时，它应设置为 2，当它是 5D 张量时，应设置为 3。

Strides

类型：const UINT*

DimensionCount 设置为 2 时，{ Depth, Height, Width } 或设置为 3 时，滑动窗口尺寸 { Height, Width } 的步幅。

WindowSize

类型：_Field_size_（DimensionCount） const UINT*

DimensionCount 设置为 2 时，{ Height, Width } 中滑动窗口的尺寸，或设置为 3 时 { Depth, Height, Width }。

StartPadding

类型：_Field_size_（DimensionCount） const UINT*

要应用于输入张量的每个空间维度的开头的填充元素数 InputTensor。 当 DimensionCount 设置为 2 或设置为 3 时 { Depth, Height, Width } 时，这些值将处于 { Height, Width } 中。

EndPadding

类型：_Field_size_（DimensionCount） const UINT*

要应用于输入张量的每个空间维度末尾的填充元素数 InputTensor。 当 DimensionCount 设置为 2 或设置为 3 时 { Depth, Height, Width } 时，这些值将处于 { Height, Width } 中。

言论

DML_MAX_POOLING1_OPERATOR_DESC中引入了较新版本的此运算符 DML_FEATURE_LEVEL_2_0。

可用性

此运算符是在 DML_FEATURE_LEVEL_1_0中引入的。

Tensor 约束

InputTensor 和 OutputTensor 必须具有相同的 DataType，DimensionCount。

Tensor 支持

DML_FEATURE_LEVEL_5_0及更高版本

张肌	类	支持的维度计数	支持的数据类型
InputTensor	输入	4 到 5	FLOAT32、FLOAT16、INT64、INT32、INT16、INT8、UINT64、UINT32、UINT16、UINT8
OutputTensor	输出	4 到 5	FLOAT32、FLOAT16、INT64、INT32、INT16、INT8、UINT64、UINT32、UINT16、UINT8

DML_FEATURE_LEVEL_3_0及更高版本

张肌	类	支持的维度计数	支持的数据类型
InputTensor	输入	4 到 5	FLOAT32、FLOAT16、INT8、UINT8
OutputTensor	输出	4 到 5	FLOAT32、FLOAT16、INT8、UINT8

DML_FEATURE_LEVEL_1_0及更高版本

张肌	类	支持的维度计数	支持的数据类型
InputTensor	输入	4 到 5	FLOAT32，FLOAT16
OutputTensor	输出	4 到 5	FLOAT32，FLOAT16

要求

要求	价值
标头	directml.h