code: def __init__(self, units=32, input_dim=32):
      super().__init__()
      self.w = self.add_weight(
          shape=(input_dim, units),
          initializer="random_normal",
          trainable=True,
      )
      self.b = self.add_weight(
          shape=(units,), initializer="zeros", trainable=True
      )
  
imports: import numpy
  import keras
  from keras import ops
  from keras import layers
label_desc: Use image node_id 1 to create a new Linear object from inherited base classes: keras.layers with example: obj = Linear(units, input_dim)