deephyper.keras.layers.AttentionConst#

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

Bases: Layer

Constant Attention.

The attention coefficient between node \(i\) and \(j\) is calculated as:

\[\alpha_{ij} = 1\]
Parameters:
  • state_dim (int) – number of output channels.

  • attn_heads (int) – number of attention heads.

Methods

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) – X (tensor): node feature tensor N (int): number of nodes targets (tensor): target node index tensor sources (tensor): source node index tensor degree (tensor): node degree sqrt tensor (for GCN attention)

Returns:

attention coefficient tensor

Return type:

attn_coef (tensor)