code: def __init__(self, units, use_bias, **kwargs):
      super(Linear, self).__init__(**kwargs)
      self.linear_layer = keras.layers.Dense(
          units, use_bias=use_bias, name="linear_layer"
      )
      self.activation = keras.layers.ReLU()
  
imports: import tensorflow
  from tensorflow import keras
  from tensorflow.python.ops import math_ops
label_desc: Use image node_id 1 to create a new Linear object from inherited base classes: keras.layers with example: obj = Linear(units, use_bias)