def get_config(self):
        config = super().get_config()
        config.update({"shape": self.shape})
        return config