code: def __init__(self, input_dim, output_size, name=None):
      super(_Dense, self).__init__(name=name)
      self.w = tf.Variable(
          tf.random.normal([input_dim, output_size]), name="w"
      )
      self.b = tf.Variable(tf.zeros([1, output_size]), name="b")
  
imports: from absl.testing import absltest
  from absl.testing import parameterized
  import jax
  from jax import numpy
  import neural_tangents
  from neural_tangents import experimental
  import numpy
  import tensorflow
label_desc: Use image node_id 1 to create a new _Dense object from inherited base classes: tf with example: obj = _Dense(input_dim, output_size, name)