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