自訂運算子
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 上提出問題。