deephyper.keras.layers.GlobalAttentionPool
deephyper.keras.layers.GlobalAttentionPool#
-
class
deephyper.keras.layers.
GlobalAttentionPool
(*args: Any, **kwargs: Any)[source]# 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
Apply the layer on input tensors.
-
__call__
(*args: Any, **kwargs: Any) → Any# Call self as a function.