Hi @albanD, sorry for the direct post but I wasn’t sure who else to tag. Is there a way to “insert a (any) GraphNN layer” in a feedforward model I already have in pytorch (e.g. pytorch geometric)?
e.g.
imagine I wanted to implement a sequential model with a GCN:
f = nn.Sequential(OrderedDict([
    ('f1', nn.Linear(Din, Dout)),
    ('out', nn.ReLU()),
    ('gcn1', torch_geometric.GCN(*params))
]))
y: Tensor =  f(x)

Something like that. Is it possible?