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