ALIGNNConv

ALIGNNConv#

class nfflr.nn.ALIGNNConv(in_features: int, out_features: int, norm: Literal['layernorm', 'batchnorm'] = 'layernorm')[source]#

Line graph update.

Methods

forward(g, lg, x, y, z)

Node and Edge updates for ALIGNN layer.

forward(g: DGLGraph, lg: DGLGraph, x: Tensor, y: Tensor, z: Tensor)[source]#

Node and Edge updates for ALIGNN layer.

x: node input features y: edge input features z: edge pair input features