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","."]
docstr: The mean of the normal distribution.
code: def loc(self):
      """The mean of the normal distribution."""
      return self._mean
  
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 4 for calling the LearnableMultivariateNormalDiag obj's underlying member method code with example usage: obj.loc() and returns: self