not_to_static¶
被本装饰器装饰的函数在动转静过程不会进行动转静代码转写。
参数¶
function (callable):装饰的函数。
返回¶
callable,一个在动转静过程不会进行代码转写的函数。
示例代码¶
import paddle
@paddle.jit.not_to_static
def func_not_to_static(x):
res = x - 1
return res
@paddle.jit.to_static
def func(x):
if paddle.mean(x) < 0:
out = func_not_to_static(x)
else:
out = x + 1
return out
x = paddle.ones([1, 2], dtype='float32')
out = func(x)
print(out) # [[2. 2.]]