问如何为Keras定义一个新的优化函数

EN

from keras.optimizers import Optimizer

class CustomOptimizer(Optimizer):

    def __init__(self, custom, parameters, **kwargs):
        self.custom = custom
        self.parameters = parameters
        # ...
        super().__init__(kwargs)

    def get_updates(self, loss, params):
        # calculate the updates for the parameters
        grads = self.get_gradients(loss, params)  # as implemented in `Optimizer`
        # ...
        self.updates = # your updates
        return self.updates

    def get_config(self):
        return {'custom': custom, 'parameters': parameters}