Updated map docstring to allow for more general input shapes.