code: def fork_rng_wrap(function=None, **kwargs):
      """Decorator alias for `fork_rng`."""
      if not function:
          return functools.partial(fork_rng_wrap, **kwargs)
  
      @functools.wraps(function)
      def wrapper():
          with fork_rng(**kwargs):
              return function()
  
      return wrapper
  
file_path: pywick/pywick/cust_random.py
imports: from collections import namedtuple
  from contextlib import contextmanager
  import functools
  import random
  import numpy
  import torch
label_desc: Use image node_id 4 for calling a global function with example usage: fork_rng_wrap(function) and returns: wrapper, functools, function