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 |