code: def rand_basis(ws: Dict, device: Optional[str] = torch.device("cpu")):
      return {
          k: torch.randn(size=v.shape, device=device)
          for k, v in ws.items()
      }
  
file_path: ml-cvnets/loss_landscape/landscape_utils.py
imports: import copy
  from typing import Dict, Optional
  import matplotlib.pyplot
  import numpy
  import torch
  from matplotlib import animation, cm
  from mpl_toolkits.mplot3d import Axes3D
  from utils.common_utils import unwrap_model_fn
label_desc: Use image node_id 1 for calling a global function with example usage: rand_basis(ws, device) and returns: unknown