deephyper.keras.layers.GlobalAttentionSumPool#

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

Bases: tensorflow.keras.layers.

Global Attention Summation Pool.

Pools a graph by learning attention coefficients to sum node features. Details can be seen from https://github.com/danielegrattarola/spektral

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 (tensor) – the node feature tensor

Returns

GlobalAttentionSumPool tensor (tensor)