infer_shape.py 37 KB