code: def vmap(
      func: Callable, in_dims: in_dims_t = 0, out_dims: out_dims_t = 0
  ) -> Callable:
      """
      Please use torch.vmap instead of this API.
      """
      warnings.warn(
          "Please use torch.vmap instead of torch._vmap_internals.vmap. ",
          stacklevel=2,
      )
      return _vmap(func, in_dims, out_dims)
  
file_path: pytorch/torch/_vmap_internals.py
imports: import functools
  import warnings
  from typing import Any, Callable, List, Optional, Tuple, Union
  import torch
  from torch import Tensor
  from torch.utils._pytree import _broadcast_to_and_flatten, tree_flatten, tree_unflatten
label_desc: Use image node_id 9 for calling a global function with example usage: vmap(func, in_dims, out_dims) and returns: _vmap