Added infer_shape into example and fixed line syntax.