IMLOperatorShapeInferrer interface

Implemented by shape inferrers to infer shapes of an operator's output tensor edges. Shape inferrers may be provided when registering custom operator kernels to improve performance and to enable the kernel to query the shape of its output tensors when it is created and computed. Shape inferrers may also be provided when registering custom operator schema to improve model validation.

Methods

Name Description
InferOutputShapes Called to infer shapes of an operator's output edges.

Requirements

Requirement
Minimum supported client Windows 10, build 17763
Minimum supported server Windows Server 2019 with Desktop Experience
Header MLOperatorAuthor.h

Note

Use the following resources for help with Windows ML:

  • To ask or answer technical questions about Windows ML, please use the windows-machine-learning tag on Stack Overflow.
  • To report a bug, please file an issue on our GitHub.