infer_shape

产品支持情况

全量芯片支持。

功能说明

给定输入shape，对传入的Graph做全图shape推导。

本接口只做shape推导，不对图做任何其他优化（如常量折叠、死边消除等）。

函数原型

1

infer_shape(graph: Graph, input_shapes: List[List[int]]) -> None

参数说明

返回值说明

无

约束说明

• 如果input_shapes不是整数列表的列表，抛出TypeError。
• 如果shape推导失败，抛出RuntimeError。