在 Win32 API 中，DML_ARGMAX_OPERATOR_DESC 结构用于描述 Argmax 运算符的参数。以下是该结构的定义：
typedef struct DML_ARGMAX_OPERATOR_DESC {
    DML_TENSOR_DESC InputTensor;
    DML_TENSOR_DESC OutputTensor;
    DML_AXIS_AXISWISE_OP Axis;
    DML_ARGMAX_MODE Mode;
} DML_ARGMAX_OPERATOR_DESC;

这个结构包含四个成员：

1. InputTensor：描述输入张量的结构（DML_TENSOR_DESC 类型）。
2. OutputTensor：描述输出张量的结构（DML_TENSOR_DESC 类型）。
3. Axis：指定 Argmax 运算的轴。可以使用 DML_AXIS_AXISWISE_OP 枚举类型来设置轴。
4. Mode：指定 Argmax 运算的模式。可以使用 DML_ARGMAX_MODE 枚举类型来设置模式。

你可以使用这个结构来配置 Argmax 运算符的输入和输出张量的参数，以及指定运算的轴和模式。需要根据具体的应用场景和需求来填充和配置这个结构。




转载请注明出处：http://www.zyzy.cn/article/detail/26830/Win32 API/Directml.h/DML_ARGMAX_OPERATOR_DESC