code: def vmap(
      func: Callable,
      in_dims: in_dims_t = 0,
      out_dims: out_dims_t = 0,
      randomness: str = "error",
      *,
      chunk_size=None,
  ) -> Callable:
      warn_deprecated("vmap", "torch.vmap")
      return apis.vmap(
          func, in_dims, out_dims, randomness, chunk_size=chunk_size
      )
  
imports: import torch._functorch.apis
  import torch._functorch.eager_transforms
  import torch._functorch.make_functional
  from torch._functorch.vmap import in_dims_t, out_dims_t
  from torch._functorch.eager_transforms import argnums_t
  import torch.nn
  import textwrap
  from typing import Any, Callable, Optional, Tuple, Union
  import warnings
label_desc: Use image node_id 4 for calling a global function with example usage: vmap(func, in_dims, out_dims, randomness) and returns: apis