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

tf_agents.utils.nest_utils.get_outer_shape

View source on GitHub

Runtime batch dims of tensor's batch dimension dim.

tf_agents.utils.nest_utils.get_outer_shape(
    nested_tensor, spec
)