class_docstr: Monitors layer calls to the underlying distribution.
class_docstr_tok: ["Monitors","layer","calls","to","the","underlying","distribution","."]
code: def scale(self):
      self.called_scale()
      return self.result_scale
  
imports: import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import independent
  from tensorflow_probability.python.distributions import normal
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.internal import tf_keras
  from tensorflow_probability.python.layers import dense_variational
  from tensorflow_probability.python.layers import util
  from tensorflow_probability.python.random import random_ops
  from tensorflow_probability.python.util import seed_stream
label_desc: Use image node_id 6 for calling the MockDistribution obj's underlying member method code with example usage: obj.scale() and returns: self