deephyper.keras.layers.GlobalAttentionPool

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

Bases: tensorflow.keras.layers.

Global Attention Pool.

A gated attention global pooling layer as presented by [Li et al. (2017)](https://arxiv.org/abs/1511.05493). Details can be seen from https://github.com/danielegrattarola/spektral

Parameters

state_dim (int) – number of output channels.

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

GlobalAttentionPool tensor (tensor)