class_docstr: Test class for the individual model components.
class_docstr_tok: ["Test","class","for","the","individual","model","components","."]
code: def testLearnableMultivariateNormalDiagClass(self):
      dist_model = disentangled_vae.LearnableMultivariateNormalDiag(
          self.dimensions
      )
      dist = dist_model()
      self.assertDistShape(dist, (self.dimensions,), ())
  
imports: import numpy
  import tensorflow.compat.v1
  from tensorflow_probability.examples import disentangled_vae
  from tensorflow.python.framework import test_util
label_desc: Use image node_id 1 for calling the DisentangledVAEComponentsTest obj's underlying member method code with example usage: obj.testLearnableMultivariateNormalDiagClass() without return types