code: def test_dnn_nobayes(self):
      affine = tfn.Affine(5, 10, batch_shape=())
      inputs = tf.zeros([4, 5])
      output = affine(inputs)
      self.assertShapeEqual(np.zeros([4, 10]), output)
  
imports: import functools
  from absl.testing import parameterized
  import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.bijectors import softmax_centered
  from tensorflow_probability.python.bijectors import softplus
  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 prefer_static
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.internal import tf_keras
  from tensorflow_probability.python.util import deferred_tensor
label_desc: Use image node_id 2 for calling the AffineTest obj's underlying member method code with example usage: obj.test_dnn_nobayes() without return types