File tree 1 file changed +5
-5
lines changed
1 file changed +5
-5
lines changed Original file line number Diff line number Diff line change @@ -7,17 +7,17 @@ to_static
7
7
8
8
本装饰器将函数内的动态图 API 转化为静态图 API。此装饰器自动处理静态图模式下的 Program 和 Executor,并将结果作为动态图 Tensor 返回。输出的动态图 Tensor 可以继续进行动态图训练、预测或其他运算。如果被装饰的函数里面调用其他动态图函数,被调用的函数也会被转化为静态图函数。
9
9
10
-
11
10
参数
12
11
::::::::::::
13
12
14
13
- **function ** (callable) - 待转换的动态图函数。若以装饰器形式使用,则被装饰函数默认会被解析为此参数值,无需显式指定。
15
14
- **input_spec ** (list[InputSpec]|tuple[InputSpec]) - 用于指定被装饰函数中输入 Tensor 的 shape、dtype 和 name 信息,为包含 InputSpec 的 list/tuple 类型。
16
- - **build_strategy ** (BuildStrategy|None):通过配置 build_strategy,对转换后的计算图进行优化,例如:计算图中算子融合、计算图执行过程中开启内存/显存优化等。关于 build_strategy 更多信息,请参阅 `` paddle.static.BuildStrategy `` 。默认为 None。
17
- - **backend ** (str,可选): 指定后端编译器,可以指定为 `CINN ` 或者 None。当该参数指定为 `CINN ` 时,将会使用 CINN 编译器来加速训练和推理。
18
- - **kwargs **: 支持的 key 包括 `property `
15
+ - **build_strategy ** (BuildStrategy|None):通过配置 :attr: ` build_strategy ` ,对转换后的计算图进行优化,例如:计算图中算子融合、计算图执行过程中开启内存/显存优化等。关于 :attr: ` build_strategy ` 更多信息,请参阅 :ref: ` paddle.static.BuildStrategy < cn_api_paddle_static_BuildStrategy >` 。默认为 `` None `` 。
16
+ - **backend ** (str,可选): 指定后端编译器,可以指定为 `` " CINN" `` 或者 `` None `` 。当该参数指定为 `` " CINN" `` 时,将会使用 `` CINN `` 编译器来加速训练和推理。
17
+ - **kwargs **: 支持的 key 包括 :attr: `property ` 和 :attr: ` full_graph `。
19
18
20
- - **property **: 表示被装饰的函数是否以 class property 属性的方式进行导出
19
+ - **property ** (bool): 表示被装饰的函数是否以 class property 属性的方式进行导出,默认为 ``False ``。
20
+ - **full_graph ** (bool): 表示被装饰的函数是否以整图静态图的方式进行导出,默认为 ``False ``。
21
21
22
22
23
23
代码示例
You can’t perform that action at this time.
0 commit comments