typedef struct DML_AVERAGE_POOLING_GRAD_OPERATOR_DESC {
DML_TENSOR_DESC InputTensor;
DML_TENSOR_DESC OutputGradientTensor;
DML_TENSOR_DESC InputGradientTensor;
DML_POOLING_KERNEL_DESC Kernel;
DML_POOLING_STRIDE_DESC Stride;
DML_POOLING_OUTPUT_PADDING_DESC OutputPadding;
UINT32 DivisionFactor;
} DML_AVERAGE_POOLING_GRAD_OPERATOR_DESC;
这个结构包含七个成员:
1. InputTensor:描述输入张量的结构(DML_TENSOR_DESC 类型)。
2. OutputGradientTensor:描述输出梯度张量的结构(DML_TENSOR_DESC 类型)。
3. InputGradientTensor:描述输入梯度张量的结构(DML_TENSOR_DESC 类型)。
4. Kernel:描述池化核的结构,包括大小和填充方式(DML_POOLING_KERNEL_DESC 类型)。
5. Stride:描述池化步幅的结构(DML_POOLING_STRIDE_DESC 类型)。
6. OutputPadding:描述输出张量的填充方式(DML_POOLING_OUTPUT_PADDING_DESC 类型)。
7. DivisionFactor:指定平均池化运算中的除数因子。
你可以使用这个结构来配置 Average Pooling 梯度运算符的输入和输出张量的参数,以及定义池化核大小、步幅、输出填充方式和除数因子等参数。
需要注意,为了正确使用这个结构,建议仔细查阅相关的 DirectML 文档,以确保正确配置参数以满足你的具体需求。
转载请注明出处:http://www.zyzy.cn/article/detail/26832/Win32 API/Directml.h/DML_AVERAGE_POOLING_GRAD_OPERATOR_DESC