code: def test_correctly_returns_non_tensor(self):
      input_size = 3
      output_size = 5
      f = AffineMeanFieldNormal(output_size, input_size)
      x = tf.zeros([2, 1, input_size])
      y = f(x)
      self.assertIsInstance(y, independent.Independent)
      self.assertAllEqual((2, 1, output_size), y.distribution.loc.shape)
  
imports: import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import independent
  from tensorflow_probability.python.distributions import normal
  from tensorflow_probability.python.experimental import nn
  from tensorflow_probability.python.internal import test_util
label_desc: Use image node_id 1 for calling the LayerTest obj's underlying member method code with example usage: obj.test_correctly_returns_non_tensor() without return types