code: def set_rng_seed(rng_seed):
      """
      Sets seeds of `torch` and `torch.cuda` (if available).
  
      :param int rng_seed: The seed value.
      """
      torch.manual_seed(rng_seed)
      random.seed(rng_seed)
      np.random.seed(rng_seed)
  
imports: import functools
  import math
  import numbers
  import random
  import sys
  import timeit
  import warnings
  from collections import defaultdict
  from contextlib import contextmanager
  from itertools import zip_longest
  import numpy
  import torch
  from pyro.poutine.util import site_is_subsample
label_desc: Use image node_id 1 for calling a global function with example usage: set_rng_seed(rng_seed) without return types