class_docstr: Learnable multivariate diagonal normal distribution.
  
  The model is a multivariate normal distribution with learnable
  `mean` and `stddev` parameters.
class_docstr_tok: ["Learnable","multivariate","diagonal","normal","distribution",".","The","model","is","a","multivariate","normal","distribution","with","learnable","`","mean","`","and","`","stddev","`","parameters","."]
code: def __call__(self, *args, **kwargs):
      # Allow this Model to be called without inputs.
      dummy = tf1.zeros(self.dimensions)
      return super(LearnableMultivariateNormalDiag, self).__call__(
          dummy, *args, **kwargs
      )
  
imports: from datetime import datetime
  import functools
  from absl import app
  from absl import flags
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  import tensorflow_probability
  from tensorflow_probability.examples import sprites_dataset
  from tensorflow_probability.python.internal import tf_keras
label_desc: Use image node_id 2 for calling the LearnableMultivariateNormalDiag obj's underlying member method code with example usage: obj.__call__() and returns: super