deephyper.keras.layers.UpdateFuncMLP#

class deephyper.keras.layers.UpdateFuncMLP(*args: Any, **kwargs: Any)[source]#

Bases: Layer

Multi-layer perceptron update function.

Parameters:
  • state_dim (int) – number of output channels.

  • activation (str) – the type of activation functions.

Methods

build

call

Apply the layer on input tensors.

__call__(*args: Any, **kwargs: Any) Any#

Call self as a function.

call(inputs, **kwargs)[source]#

Apply the layer on input tensors.

Parameters:

inputs (list) – old_state (tensor): node hidden feature tensor agg_messages (tensor): node hidden feature tensor

Returns:

activated tensor from update function (

Return type:

activation (tensor)