共用方式為


自訂運算子

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 上提出問題。