typedef struct DML_GEMM_OPERATOR_DESC {
const DML_TENSOR_DESC *ATensor;
const DML_TENSOR_DESC *BTensor;
const DML_TENSOR_DESC *CTensor;
const DML_TENSOR_DESC *DTensor;
DML_MATRIX_TRANSFORM Transform;
DML_MATRIX_TRANSFORM AccumulationMode;
DML_ACTIVATION_DESC FusedActivation;
BOOL FusedActivationValid;
BOOL BiasTensorValid;
BOOL BScaleValid;
BOOL CScaleValid;
BOOL ZeroCValid;
const DML_TENSOR_DESC *BiasTensor;
float BScale;
float CScale;
float ZeroC;
} DML_GEMM_OPERATOR_DESC;
这个结构包含了一些输入张量(ATensor、BTensor)、输出张量(CTensor、DTensor)的描述,以及一些其他的参数,如矩阵变换、累积模式、激活函数等。这里还包括了一些用于优化的标志(如 FusedActivationValid、BiasTensorValid 等)以及相应的张量描述(BiasTensor、BScale、CScale等)。
为了获取最新的和详细的信息,建议查阅 Microsoft 官方的 DirectML 文档或直接查看最新版本的 Directml.h 头文件。
转载请注明出处:http://www.pingtaimeng.com/article/detail/26854/Win32 API/Directml.h/DML_GEMM_OPERATOR_DESC