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