ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

tf_agents.policies.policy_saver.add_batch_dim