自訂運算子

Windows Machine Learning 自訂運算子 WIN32 API 位於 MLOperatorAuthor.h中。

API

以下是具有其語法和描述的自訂運算子 API 清單。

列舉

名稱 描述
MLOperatorAttributeType 指定屬性的類型。 每個屬性類型會以數值方式比對對應的 ONNX 類型。
MLOperatorEdgeType 指定運算子之輸入或輸出邊緣的類型。
MLOperatorExecutionType 指定核心是否使用 CPU 或 GPU 進行計算。
MLOperatorKernelOptions 指定註冊自訂運算子核心時所使用的選項。
MLOperatorParameterOptions 指定運算子輸入和輸出邊緣的選項旗標。
MLOperatorSchemaEdgeTypeFormat 指定描述輸入和輸出邊緣類型的方式。
MLOperatorTensorDataType 指定 Tensor 的資料類型。 每個資料類型會以數值方式比對對應的 ONNX 類型。

函式

名稱 描述
MLCreateOperatorRegistry 建立 IMLOperatorRegistry 的實例,此實例可用來註冊自訂運算子核心和自訂運算子架構。

介面

名稱 描述
IMLOperatorAttributes 表示運算子屬性的值,如使用 運算子的模型所決定。
IMLOperatorKernel 由自訂運算子核心實作。
IMLOperatorKernelContext 提供在計算核心時運算子使用方式的相關資訊。
IMLOperatorKernelCreationContext 提供在建立核心時操作員使用方式的相關資訊。
IMLOperatorKernelFactory 由自訂運算子核心的作者實作,以建立該核心的實例。
IMLOperatorRegistry 表示自訂運算子核心和架構之登錄的實例。
IMLOperatorShapeInferenceContext 提供叫用圖形推斷器時運算子使用方式的相關資訊。
IMLOperatorShapeInferrer 由圖形推斷器實作,以推斷運算子輸出張量邊緣的圖形。
IMLOperatorTensor 在計算自訂運算子核心期間使用的 Tensor 標記法。
IMLOperatorTensorShapeDescription 表示運算子的輸入和輸出張量圖形集。
IMLOperatorTypeInferenceContext 提供在叫用類型推斷器時,運算子使用方式的相關資訊。
IMLOperatorTypeInferrer 由類型推斷器實作,以推斷運算子的輸出邊緣類型。

結構

名稱 描述
MLOperatorAttribute 指定自訂運算子之屬性的名稱和屬性。
MLOperatorAttributeNameValue 指定自訂運算子屬性的名稱和值 () 。
MLOperatorEdgeDescription 指定運算子之輸入或輸出邊緣的屬性。
MLOperatorEdgeTypeConstraint 指定自訂運算子核心和架構所支援邊緣類型的條件約束。
MLOperatorKernelDescription 用來註冊該架構的自訂運算子核心描述。
MLOperatorSchemaDescription 用來註冊該架構的自訂運算子架構描述。
MLOperatorSchemaEdgeDescription 指定運算子之輸入或輸出邊緣的相關資訊。
MLOperatorSetId 指定操作員集的身分識別。

注意

使用下列資源取得 Windows ML 的說明:

  • 如需詢問或回答有關 Windows ML 的技術問題,請使用 Stack Overflow 上的 windows-machine-learning 標籤。
  • 如需回報錯誤 (bug),請在 GitHub 上提出問題。