在 Win32 API 中,DML_AVERAGE_POOLING_OPERATOR_DESC 结构用于描述 Average Pooling 运算符的参数。以下是该结构的定义:
typedef struct DML_AVERAGE_POOLING_OPERATOR_DESC {
    DML_TENSOR_DESC InputTensor;
    DML_TENSOR_DESC OutputTensor;
    DML_POOLING_KERNEL_DESC Kernel;
    DML_POOLING_STRIDE_DESC Stride;
    DML_POOLING_OUTPUT_PADDING_DESC OutputPadding;
    DML_POOLING_INPUT_PADDING_DESC InputPadding;
    UINT32 DivisionFactor;
} DML_AVERAGE_POOLING_OPERATOR_DESC;

这个结构包含七个成员:

1. InputTensor:描述输入张量的结构(DML_TENSOR_DESC 类型)。
2. OutputTensor:描述输出张量的结构(DML_TENSOR_DESC 类型)。
3. Kernel:描述池化核的结构,包括大小和填充方式(DML_POOLING_KERNEL_DESC 类型)。
4. Stride:描述池化步幅的结构(DML_POOLING_STRIDE_DESC 类型)。
5. OutputPadding:描述输出张量的填充方式(DML_POOLING_OUTPUT_PADDING_DESC 类型)。
6. InputPadding:描述输入张量的填充方式(DML_POOLING_INPUT_PADDING_DESC 类型)。
7. DivisionFactor:指定平均池化运算中的除数因子。

你可以使用这个结构来配置 Average Pooling 运算符的输入和输出张量的参数,以及定义池化核大小、步幅、输入和输出填充方式以及除数因子等参数。

需要注意,为了正确使用这个结构,建议仔细查阅相关的 DirectML 文档,以确保正确配置参数以满足你的具体需求。




转载请注明出处:http://www.zyzy.cn/article/detail/26833/Win32 API/Directml.h/DML_AVERAGE_POOLING_OPERATOR_DESC