code: def setUp(self):
      super(WeightNormTest, self).setUp()
      self.data_dim = 2
      self.batch_size = 16
      self.num_hidden = 5
      self.random_input = np.random.rand(
          self.batch_size, self.data_dim
      ).astype(np.float32)
      self.random_targets = np.random.rand(self.batch_size, 1).astype(
          np.float32
      )
  
      data_dim = (32, 32, 3)
      self.conv_random_input = np.random.rand(
          self.batch_size, *data_dim
      ).astype(np.float32)
      self.num_conv_filters = 5
  
imports: import tempfile
  from absl.testing import parameterized
  import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.internal import tf_keras
  from tensorflow_probability.python.layers import weight_norm
label_desc: Use image node_id 1 for calling the WeightNormTest obj's underlying member method code with example usage: obj.setUp() without return types