typedef struct DML_CONVOLUTION_OPERATOR_DESC {
DML_TENSOR_DESC InputTensor;
DML_TENSOR_DESC FilterTensor;
DML_TENSOR_DESC BiasTensor;
DML_TENSOR_DESC OutputTensor;
UINT32 DimensionCount;
UINT32 FilterCount;
UINT32 Stride;
UINT32 Dilation;
INT32 StartPadding;
INT32 EndPadding;
DML_CONVOLUTION_MODE Mode;
BOOL Transpose;
BOOL FusedActivation;
DML_OPERATOR FusedActivationCode;
} DML_CONVOLUTION_OPERATOR_DESC;
结构成员说明:
- InputTensor:输入张量的描述。
- FilterTensor:滤波器(卷积核)的描述。
- BiasTensor:偏置张量的描述。
- OutputTensor:输出张量的描述。
- DimensionCount:张量的维度数。
- FilterCount:滤波器的数量。
- Stride:卷积操作的步幅。
- Dilation:卷积操作的膨胀率。
- StartPadding 和 EndPadding:卷积操作的开始和结束填充。
- Mode:卷积模式,是一个 DML_CONVOLUTION_MODE 枚举值,表示卷积操作的模式。
- Transpose:一个布尔值,表示是否进行转置卷积操作。
- FusedActivation:一个布尔值,表示是否包含融合的激活函数。
- FusedActivationCode:融合激活函数的代码。
请注意,上述信息是基于我截至2022年1月的知识,如果有关于 DML_CONVOLUTION_OPERATOR_DESC 的最新信息,建议查阅 Microsoft 的官方文档或者最新的 DirectML 头文件。
转载请注明出处:http://www.zyzy.cn/article/detail/26840/Win32 API/Directml.h/DML_CONVOLUTION_OPERATOR_DESC