def __init__(self, output_dim=None, activation=None, **kwargs):
self.output_dim = output_dim
self.activation = activations.get(activation)
super().__init__(**kwargs)
微信
公众号
扫码关注公众号