diff --git a/.gitignore b/.gitignore index 4dc0bf4b981cf3b52c838fefe54f4dfd3608eedc..5e1669b687d12ede6f94a5033432c1dcc799c299 100644 --- a/.gitignore +++ b/.gitignore @@ -2,3 +2,4 @@ /doc/fluid/menu.zh.json /doc/fluid/menu.en.json .idea +build diff --git a/doc/fluid/advanced_usage/deploy/inference/python_infer_cn.md b/doc/fluid/advanced_usage/deploy/inference/python_infer_cn.md index 098504f18def8a11319b11c5d0219d0ff31a3985..9e28155562925eead5f1c220695fcfef376ddeee 100644 --- a/doc/fluid/advanced_usage/deploy/inference/python_infer_cn.md +++ b/doc/fluid/advanced_usage/deploy/inference/python_infer_cn.md @@ -174,6 +174,7 @@ input_tensor = predictor.get_input_tensor(input_names[0]) # 设置输入 fake_input = numpy.random.randn(1, 3, 318, 318).astype("float32") +input_tensor.reshape([1, 3, 318, 318]) input_tensor.copy_from_cpu(fake_input) # 运行predictor @@ -190,6 +191,11 @@ output_tensor = predictor.get_output_tensor(output_names[0]) * ZeroCopyTensor * `copy_from_cpu(input: numpy.ndarray) -> None` * `copy_to_cpu() -> numpy.ndarray` + * `reshape(input: numpy.ndarray|List[int]) -> None` + * `shape() -> List[int]` + * `set_lod(input: numpy.ndarray|List[List[int]]) -> None` + * `lod() -> List[List[int]]` + * `type() -> PaddleDType` * AnalysisConfig * `set_model(model_dir: str) -> None` * `set_model(prog_file: str, params_file: str) -> None` @@ -312,6 +318,7 @@ def main(): # 设置输入 fake_input = np.random.randn(1, 3, 318, 318).astype("float32") + input_tensor.reshape([1, 3, 318, 318]) input_tensor.copy_from_cpu(fake_input) # 运行predictor