code: def nn_predict_gaussian(params, inputs):
      # Returns means and diagonal variances
      return unpack_gaussian_params(neural_net_predict(params, inputs))
  
imports: from __future__ import absolute_import, division
  from __future__ import print_function
  import autograd.numpy
  import autograd.numpy.random
  import autograd.scipy.stats.norm
  from autograd.scipy.special import expit
  from autograd import grad
  from autograd.misc.optimizers import adam
  from data import load_mnist, save_images
label_desc: Use image node_id 9 for calling a global function with example usage: nn_predict_gaussian(params, inputs) and returns: unpack_gaussian_params