Windows 机器学习自定义运算符 Win32 API 位于 MLOperatorAuthor.h 中。
应用程序接口
下面是包含其语法和说明的自定义运算符 API 的列表。
枚举
| 名称 | DESCRIPTION |
|---|---|
| MLOperatorAttributeType | 指定属性的类型。 每个属性类型以数字方式匹配相应的 ONNX 类型。 |
| MLOperatorEdgeType | 指定运算符的输入或输出边缘的类型。 |
| MLOperatorExecutionType | 指定内核是使用 CPU 还是 GPU 进行计算。 |
| MLOperatorKernelOptions | 指定注册自定义运算符内核时使用的选项。 |
| MLOperatorParameterOptions | 指定运算符输入和输出边的选项标志。 |
| MLOperatorSchemaEdgeTypeFormat | 指定描述输入和输出边缘类型的方式。 |
| MLOperatorTensorDataType | 指定张量的数据类型。 每个数据类型以数字方式匹配相应的 ONNX 类型。 |
功能
| 名称 | DESCRIPTION |
|---|---|
| MLCreateOperatorRegistry | 创建 IMLOperatorRegistry 的实例，该实例可用于注册自定义作员内核和自定义运算符架构。 |
接口
| 名称 | DESCRIPTION |
|---|---|
| IMLOperatorAttributes | 表示由使用运算符的模型确定的运算符属性的值。 |
| IMLOperatorKernel | 由自定义运算符内核实现。 |
| IMLOperatorKernelContext | 提供有关内核计算时操作符使用情况的信息。 |
| IMLOperatorKernelCreationContext | 提供有关在创建内核时操作符使用情况的信息。 |
| IMLOperatorKernelFactory | 由自定义运算符内核的作者实现，以创建该内核的实例。 |
| IMLOperatorRegistry | 表示自定义运算符内核和架构的注册表实例。 |
| IMLOperatorShapeInferenceContext | 提供有关正在调用形状推理器时运算符用法的信息。 |
| IMLOperatorShapeInferrer | 由形状推断器实现，用于推断运算符的输出张量边缘的形状。 |
| IMLOperatorTensor | 自定义运算符内核计算期间使用的张量表示形式。 |
| IMLOperatorTensorShapeDescription | 表示运算符的输入和输出张量形状集。 |
| IMLOperatorTypeInferenceContext | 提供有关调用类型推理器时运算符用法的信息。 |
| IMLOperatorTypeInferrer | 由类型推理程序实现，用于推断运算符的输出边缘的类型。 |
结构
| 名称 | DESCRIPTION |
|---|---|
| MLOperatorAttribute | 指定自定义运算符的属性的名称和属性。 |
| MLOperatorAttributeNameValue | 指定自定义运算符的属性的名称和值。 |
| MLOperatorEdgeDescription | 指定运算符的输入或输出边缘的属性。 |
| 机器学习操作符边类型约束 | 指定自定义运算符内核和架构中支持的边缘类型的约束。 |
| MLOperatorKernelDescription | 用于注册该架构的自定义运算符内核的说明。 |
| MLOperatorSchemaDescription | 用于注册该架构的自定义运算符架构的说明。 |
| MLOperatorSchemaEdgeDescription | 指定有关运算符的输入或输出边缘的信息。 |
| MLOperatorSetId | 指定运算符集合的标识。 |
注释
使用以下资源获取有关 Windows ML 的帮助：
- 若要提出或回答有关 Windows ML 的技术问题，请在 Stack Overflow 上使用 windows-machine-learning 标记。
- 若要报告 bug，请在 GitHub 上提交问题。