Transform

class deeplay.components.gnn.mpn.transformation.Transform(*args, **kwargs)

Bases: CombineLayerActivation

Transform module for MPN.

Methods Summary

get_forward_args(x)

Get the arguments for the Transform module.

Methods Documentation

get_forward_args(x)

Get the arguments for the Transform module. An MPN Transform module takes the following arguments: - node features of sender nodes (x[A[0]]) - node features of receiver nodes (x[A[1]]) - edge features (edgefeat) A is the adjacency matrix of the graph.