IMLOperatorShapeInferrer 接口

由形状推断器实现,用于推断运算符的输出张量边缘的形状。 在注册自定义运算符内核时,可以提供形状推理程序,以提高性能,并允许内核在创建和计算时查询其输出张量的形状。 注册自定义运算符架构以提高模型验证时,还可以提供形状推理程序。

方法

名称 DESCRIPTION
InferOutputShapes (推断输出形状) 调用以推断运算符的输出边缘的形状。

要求

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

注释

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

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