code: def rand_tensor(shape, mean, sigma):
      return mean * torch.ones(shape) + sigma * torch.randn(shape)
  
imports: import argparse
  import errno
  import os
  import numpy
  import torch
  import wget
  from torch.nn.functional import softplus
  import pyro
  import pyro.optim
  from pyro.contrib.easyguide import EasyGuide
  from pyro.contrib.examples.util import get_data_directory
  from pyro.distributions import Gamma, Normal, Poisson
  from pyro.infer import SVI, TraceMeanField_ELBO
  from pyro.infer.autoguide import AutoDiagonalNormal, init_to_feasible
label_desc: Use image node_id 1 for calling a global function with example usage: rand_tensor(shape, mean, sigma) and returns: unknown