enable_to_static¶

paddle.jit. enable_to_static ( enable_to_static_bool ) [source]

Enable or disable the converting from imperative to static graph by ProgramTranslator globally.

Parameters

enable_to_static_bool (bool) – True or False to enable or disable converting to static.

Returns

None.

Examples

import paddle


@paddle.jit.to_static
def func(x):
    if paddle.mean(x) > 0:
        x_v = x - 1
    else:
        x_v = x + 1
    return x_v


paddle.jit.enable_to_static(False)

x = paddle.ones([1, 2])
# ProgramTranslator is disabled so the func is run in dygraph
print(func(x))  # [[0. 0.]]