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

tf_agents.utils.nest_utils.get_outer_array_shape

View source on GitHub

Batch dims of array's batch dimension dim.

tf_agents.utils.nest_utils.get_outer_array_shape(
    nested_array, spec
)