code: def build(self, input_shape):
      self.w = self.add_weight(
          shape=(input_shape[-1], self.units),
          initializer="random_normal",
          trainable=True,
      )
      self.b = self.add_weight(
          shape=(self.units,),
          initializer="random_normal",
          trainable=True,
      )
  
imports: import numpy
  import keras
  from keras import layers
  from keras import ops
label_desc: Use image node_id 2 for calling the CustomDense obj's underlying member method code with example usage: obj.build(input_shape) without return types