IMLOperatorShapeInferrer 接口

由形状推理器实现,用于推断运算符的输出张量边缘的形状。 注册自定义运算符内核时可以提供形状推理器,以提高性能并使内核能够在创建和计算输出张量时查询其输出张量的形状。 注册自定义运算符模式时也可以提供形状推理器,以改进模型验证。

方法

名称 描述
InferOutputShapes 在推断运算符的输出边缘形状时调用。

要求

要求
最低受支持的客户端 Windows 10 内部版本 17763
支持的最低服务器版本 具有桌面体验的 Windows Server 2019
页眉 MLOperatorAuthor.h

注意

使用以下资源可获取有关 Windows ML 的帮助:

  • 若要提出或回答有关 Windows ML 的技术问题,请在 Stack Overflow 上使用 windows-machine-learning 标记。
  • 若要报告 bug,请在 GitHub 上提交问题。