typedef struct DML_SPLIT_OPERATOR_DESC {
DML_TENSOR_DESC InputTensor;
UINT32 OutputCount;
DML_TENSOR_DESC* OutputTensors;
UINT32 Axis;
} DML_SPLIT_OPERATOR_DESC;
这个结构包含了输入张量的描述,输出张量的数量以及每个输出张量的描述数组。Axis 表示在哪个轴上进行拆分操作。
请注意,这只是一个示例,具体的定义可能在不同的 DirectML 版本中有所变化。为了获取确切的信息,请查阅最新版本的 DirectML 文档或检查最新版本的 Windows SDK 中的相关头文件和文档。你可以在 Microsoft 的官方文档网站上找到相关信息,或者使用 Visual Studio 中的 Object Browser 查看 SDK 文档。
转载请注明出处:http://www.zyzy.cn/article/detail/26863/Win32 API/Directml.h/DML_SPLIT_OPERATOR_DESC