deephyper.keras.layers.GlobalAttentionSumPool

class deephyper.keras.layers.GlobalAttentionSumPool(*args: Any, **kwargs: Any)

Bases: Layer

Global Attention Summation Pool.

Pools a graph by learning attention coefficients to sum node features. Details can be seen from 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)

Apply the layer on input tensors.

Parameters:

inputs (tensor) – the node feature tensor

Returns:

GlobalAttentionSumPool tensor (tensor)