Windows Machine Learning 自定義運算符 Win32 API 位於 MLOperatorAuthor.h 中。
應用程式介面(API)
以下是具有其語法和描述的自定義運算元 API 清單。
列舉項目
| 名稱 | 說明 |
|---|---|
| MLOperatorAttributeType | 指定屬性的類型。 每個屬性類型會以數值方式比對對應的 ONNX 類型。 |
| MLOperatorEdgeType | 指定運算子的輸入或輸出邊緣類型。 |
| ML運算符執行類型 | 指定核心是否使用 CPU 或 GPU 進行計算。 |
| MLOperatorKernelOptions | 指定註冊自定義運算子核心時所使用的選項。 |
| MLOperatorParameterOptions | 指定運算元輸入和輸出邊緣的選項旗標。 |
| MLOperatorSchemaEdgeTypeFormat | 指定描述輸入和輸出邊緣類型的方式。 |
| MLOperatorTensorDataType | 指定張量的資料類型。 每個數據類型會以數值方式比對對應的 ONNX 類型。 |
功能
| 名稱 | 說明 |
|---|---|
| MLCreateOperatorRegistry | 建立 IMLOperatorRegistry 的實例,可用來註冊自定義作員核心和自定義運算符架構。 |
介面
| 名稱 | 說明 |
|---|---|
| IMLOperator屬性 | 運算子的屬性值由使用該運算子的模型所決定。 |
| IMLOperatorKernel 內核 | 由自定義運算子核心實作。 |
| IMLOperatorKernelContext | 提供關於運算核心計算時運算元使用方式的資訊。 |
| IMLOperatorKernelCreationContext | 提供有關運算子在核心創建過程中使用情況的信息。 |
| IMLOperatorKernelFactory | 由自定義作員核心的作者實作,以建立該核心的實例。 |
| IMLOperator註冊表 | 表示自訂運算核心和結構的登錄實例。 |
| IMLOperatorShapeInferenceContext | 提供叫用圖形推斷器時運算元使用方式的相關信息。 |
| IMLOperatorShapeInferrer | 由圖形推斷器實作,以推斷運算符輸出張量邊緣的圖形。 |
| IMLOperatorTensor | 在計算自定義運算子核心期間使用的張量表示法。 |
| IMLOperatorTensorShapeDescription | 表示運算子的輸入和輸出張量圖形集合。 |
| IMLOperatorTypeInferenceContext | 提供有關運算子在型別推斷程式被調用時的使用資訊。 |
| IMLOperatorTypeInferrer | 由類型推斷器實作,以推斷運算子的輸出邊緣類型。 |
結構體系
| 名稱 | 說明 |
|---|---|
| MLOperatorAttribute | 指定自定義運算子屬性的名稱和屬性。 |
| MLOperatorAttributeNameValue | 指定自訂運算子屬性的名稱和值。 |
| MLOperatorEdge描述 | 指定運算子之輸入或輸出邊緣的屬性。 |
| MLOperatorEdgeTypeConstraint | 指定自定義運算子核心和架構所支援邊緣類型的條件約束。 |
| MLOperatorKernelDescription | 用來註冊該架構的自定義運算元核心描述。 |
| MLOperatorSchemaDescription | 用來註冊該架構的自定義運算符架構描述。 |
| MLOperatorSchemaEdgeDescription | 指定運算子之輸入或輸出邊緣的相關信息。 |
| MLOperatorSetId | 指定運算集的標識。 |
備註
使用以下資源以獲得 Windows ML 的協助。
- 若要詢問或回答有關 Windows ML 的技術問題,請使用 Stack Overflow 上的 windows-machine-learning 標籤。
- 若要回報錯誤,請在 GitHub 上提出問題。