code: def test_scale_gaussian_prior(self):
      with tf.Graph().as_default():
          rng = np.random.RandomState(0)
          img_shape = (16, 2, 2, 2)
          x_rand = np.asarray(
              rng.randint(0, 10, img_shape), dtype=np.float32
          )
          z_rand = np.asarray(
              rng.randint(0, 10, img_shape), dtype=np.float32
          )
          x_t = tf.convert_to_tensor(x_rand)
          z_t = tf.convert_to_tensor(z_rand)
          dist = glow_ops.scale_gaussian_prior(
              "scale_gaussian_prior", z_t, x_t, trainable=True
          )
          with tf.Session() as sess:
              sess.run(tf.global_variables_initializer())
              mean, scale = sess.run([dist.loc, dist.scale])
              self.assertTrue(np.allclose(mean, z_rand))
              self.assertTrue(np.allclose(scale, 1.0))
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import os
  import tempfile
  from absl.testing import parameterized
  import numpy
  from six.moves import range
  from six.moves import zip
  from tensor2tensor.models.research import glow
  from tensor2tensor.models.research import glow_ops
  from tensor2tensor.utils import contrib
  from tensor2tensor.utils import hparam
  import tensorflow.compat.v1
  from tensorflow.compat.v1 import estimator
label_desc: Use image node_id 14 for calling the GlowOpsTest obj's underlying member method code with example usage: obj.test_scale_gaussian_prior() without return types