deephyper.layers

deephyper.layersΒΆ

Classes

AttentionCOS

COS Attention.

AttentionConst

Constant Attention.

AttentionGAT

GAT Attention.

AttentionGCN

GCN Attention.

AttentionGenLinear

Generalized Linear Attention.

AttentionLinear

Linear Attention.

AttentionSymGAT

GAT Symmetry Attention.

GlobalAttentionPool

Global Attention Pool.

GlobalAttentionSumPool

Global Attention Summation Pool.

GlobalAvgPool

Global Average Pool.

GlobalMaxPool

Global Max Pool.

GlobalSumPool

Global Summation Pool.

MessagePasserNNM

Message passing kernel.

MessagePassing

Message passing layer.

Padding

Multi-dimensions padding layer.

SparseMPNN

Message passing cell.

UpdateFuncGRU

Gated recurrent unit update function.

UpdateFuncMLP

Multi-layer perceptron update function.