human
I wrote the below code

    def _strip_batch_and_convert_to_shape(tensor):
        return tensor[0].get_shape()
    return nest.map_structure(_strip_batch_and_convert_to_shape, tensor_nest)

What's a good function header?
assistant
#Here's a good function header.

def _get_shape_without_batch_dimension(tensor_nest):