guides/making_new_layers_and_models_via_subclassing.py/ComputeSum/__init__
class ComputeSum:    def __init__(self, input_dim):
        super().__init__()
        self.total = self.add_weight(
            initializer="zeros", shape=(input_dim,), trainable=False
        )