typedef struct DML_ELEMENT_WISE_MULTIPLY_OPERATOR_DESC
{
DML_TENSOR_DATA_TYPE InputTensorDataType;
DML_TENSOR_DATA_TYPE OutputTensorDataType;
DML_ELEMENT_WISE_MULTIPLY_OPERATOR InputTensorDesc;
DML_ELEMENT_WISE_MULTIPLY_OPERATOR OutputTensorDesc;
DML_OPERATOR_DESC OperatorDesc;
} DML_ELEMENT_WISE_MULTIPLY_OPERATOR_DESC;
该结构包含以下字段:
- InputTensorDataType: 输入张量的数据类型。
- OutputTensorDataType: 输出张量的数据类型。
- InputTensorDesc: 输入张量的乘法运算符描述。
- OutputTensorDesc: 输出张量的乘法运算符描述。
- OperatorDesc: 通用的操作符描述,包含了与运算符有关的其他信息。
这个结构体用于定义元素逐个乘法运算符的特定属性,以便在 DirectML 图中进行相应的操作。你可以使用这个结构体来创建乘法运算符节点,并将其添加到 DirectML 图中,从而构建深度学习模型。
请注意,这只是结构体的定义,实际使用时需要与其他函数和结构体一起使用,例如创建运算符节点并将其添加到图中的相关 API。
转载请注明出处:http://www.zyzy.cn/article/detail/26851/Win32 API/Directml.h/DML_ELEMENT_WISE_MULTIPLY_OPERATOR_DESC