def __init__(self, return_selection=False, **kwargs): # kwargs for the Layer class are handled automatically super().__init__( **{k: v for k, v in kwargs.items() if is_keras_kwarg(k)}) self.supports_masking = True self.return_selection = return_selection # *_regularizer, *_constraint, *_initializer, activation, and use_bias # are dealt with automatically if passed to the constructor self.kwargs_keys = [] for key in kwargs: if is_layer_kwarg(key): attr = kwargs[key] attr = deserialize_kwarg(key, attr) self.kwargs_keys.append(key) setattr(self, key, attr) # Signature of the SRC functions self.sel_signature = inspect.signature(self.select).parameters self.red_signature = inspect.signature(self.reduce).parameters self.con_signature = inspect.signature(self.connect).parameters self.i_red_signature = inspect.signature(self.reduce_index).parameters self._n_nodes = None
def __init__(self, **kwargs): super().__init__(**{k: v for k, v in kwargs.items() if is_keras_kwarg(k)}) self.kwargs_keys = [] for key in kwargs: if is_layer_kwarg(key): attr = kwargs[key] attr = deserialize_kwarg(key, attr) self.kwargs_keys.append(key) setattr(self, key, attr)
def __init__(self, **kwargs): super().__init__(**{k: v for k, v in kwargs.items() if is_keras_kwarg(k)}) self.supports_masking = True self.kwargs_keys = [] for key in kwargs: if is_layer_kwarg(key): attr = kwargs[key] attr = deserialize_kwarg(key, attr) self.kwargs_keys.append(key) setattr(self, key, attr) self.call = check_dtypes_decorator(self.call)
def __init__(self, aggregate="sum", **kwargs): super().__init__(**{k: v for k, v in kwargs.items() if is_keras_kwarg(k)}) self.kwargs_keys = [] for key in kwargs: if is_layer_kwarg(key): attr = kwargs[key] attr = deserialize_kwarg(key, attr) self.kwargs_keys.append(key) setattr(self, key, attr) self.msg_signature = inspect.signature(self.message).parameters self.agg_signature = inspect.signature(self.aggregate).parameters self.upd_signature = inspect.signature(self.update).parameters self.agg = deserialize_scatter(aggregate)