code: def log_prob(dist, batch):
      return dist.log_prob(batch)
  
file_path: pyro/profiler/distributions.py
imports: import argparse
  import torch
  from torch.autograd import Variable
  from profiler.profiling_utils import Profile, profile_print
  from pyro.distributions import Bernoulli, Beta, Categorical, Cauchy, Dirichlet, Exponential, Gamma, LogNormal, Normal, OneHotCategorical, Poisson, Uniform
label_desc: Use image node_id 5 for calling a global function with example usage: log_prob(dist, batch) and returns: dist