class_docstr: Test class for the DisentangledSequentialVAE model.
class_docstr_tok: ["Test","class","for","the","DisentangledSequentialVAE","model","."]
code: def setUp(self):
      super(DisentangledSequentialVAETest, self).setUp()
      self.latent_size_static = 10
      self.latent_size_dynamic = 11
      self.model_factorized = (
          disentangled_vae.DisentangledSequentialVAE(
              self.latent_size_static,
              self.latent_size_dynamic,
              self.hidden_size,
              self.channels,
              "factorized",
          )
      )
      self.model_full = disentangled_vae.DisentangledSequentialVAE(
          self.latent_size_static,
          self.latent_size_dynamic,
          self.hidden_size,
          self.channels,
          "full",
      )
      self.inputs = tf.random.normal(
          [self.batch_size, self.length, 64, 64, self.channels]
      )
  
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 DisentangledSequentialVAETest obj's underlying member method code with example usage: obj.setUp() without return types