typedef struct DML_ARGMIN_OPERATOR_DESC {
DML_TENSOR_DESC InputTensor;
DML_TENSOR_DESC OutputTensor;
DML_AXIS_AXISWISE_OP Axis;
DML_ARGMIN_MODE Mode;
} DML_ARGMIN_OPERATOR_DESC;
这个结构包含四个成员:
1. InputTensor:描述输入张量的结构(DML_TENSOR_DESC 类型)。
2. OutputTensor:描述输出张量的结构(DML_TENSOR_DESC 类型)。
3. Axis:指定 Argmin 运算的轴。可以使用 DML_AXIS_AXISWISE_OP 枚举类型来设置轴。
4. Mode:指定 Argmin 运算的模式。可以使用 DML_ARGMIN_MODE 枚举类型来设置模式。
你可以使用这个结构来配置 Argmin 运算符的输入和输出张量的参数,以及指定运算的轴和模式。需要根据具体的应用场景和需求来填充和配置这个结构。
转载请注明出处:http://www.zyzy.cn/article/detail/26831/Win32 API/Directml.h/DML_ARGMIN_OPERATOR_DESC