Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

tf_agents.utils.common.function

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.