View source on GitHub |
Wrapper for tf.function with TF Agents-specific customizations.
tf_agents.utils.common.function(
*args, **kwargs
)
Used in the notebooks
Used in the tutorials |
---|
Example:
@common.function()
def my_eager_code(x, y):
...
Args | |
---|---|
*args
|
Args for tf.function. |
**kwargs
|
Keyword args for tf.function. |
Returns | |
---|---|
A tf.function wrapper. |