Index

A

add_self_loops() (in module jraphx.utils)

aggregate() (MessagePassing method) (SAGEConv method)

B

BasicGNN (class in jraphx.nn.models)

BatchNorm (class in jraphx.nn.norm)

C

coalesce() (in module jraphx.utils)

D

degree() (in module jraphx.utils)

DynamicEdgeConv (class in jraphx.nn.conv)

E

EdgeConv (class in jraphx.nn.conv)

G

GAT (class in jraphx.nn.models) GATConv (class in jraphx.nn.conv) GATv2Conv (class in jraphx.nn.conv) GCN (class in jraphx.nn.models) gcn_norm() (GCNConv method) GCNConv (class in jraphx.nn.conv) GIN (class in jraphx.nn.models)

GINConv (class in jraphx.nn.conv) global_add_pool() (in module jraphx.nn.pool) global_max_pool() (in module jraphx.nn.pool) global_mean_pool() (in module jraphx.nn.pool) global_min_pool() (in module jraphx.nn.pool) GraphNorm (class in jraphx.nn.norm) GraphSAGE (class in jraphx.nn.models)

I

in_features (MLP property) init_conv() (BasicGNN method) (GAT method) (GCN method) (GIN method) (GraphSAGE method)

J

jraphx module

JumpingKnowledge (class in jraphx.nn.models)

L

LayerNorm (class in jraphx.nn.norm)

M

message() (EdgeConv method) (GINConv method) (MessagePassing method) (SAGEConv method) (TransformerConv method)

message_and_aggregate() (MessagePassing method) MessagePassing (class in jraphx.nn.conv) MLP (class in jraphx.nn.models) module jraphx

N

num_layers (MLP property)

O

out_features (MLP property)

P

propagate() (MessagePassing method)

R

remove_self_loops() (in module jraphx.utils)

reset_cache() (GCNConv method)

S

SAGEConv (class in jraphx.nn.conv) SAGPooling (class in jraphx.nn.pool) scatter() (in module jraphx.utils.scatter)

scatter_max() (in module jraphx.utils.scatter) scatter_mean() (in module jraphx.utils.scatter) scatter_min() (in module jraphx.utils.scatter) scatter_std() (in module jraphx.utils.scatter)

T

to_dense_adj() (in module jraphx.utils) to_edge_index() (in module jraphx.utils)

to_undirected() (in module jraphx.utils) TopKPooling (class in jraphx.nn.pool) TransformerConv (class in jraphx.nn.conv)

U

update() (MessagePassing method)