promptbind.models.egnn

Most codes are copied from https://github.com/vgsatorras/egnn, which is the official implementation of the paper:

E(n) Equivariant Graph Neural Networks Victor Garcia Satorras, Emiel Hogeboom, Max Welling

Functions

coord2radial(edge_index, coord, rm_F_norm[, ...])
get_edges(n_nodes)
get_edges_batch(n_nodes, batch_size)
unsorted_segment_mean(data, segment_ids, ...)
unsorted_segment_sum(data, segment_ids, ...)

Classes

MCAttEGNN(args, in_node_nf, hidden_nf, ...)
MC_Att_L(args, input_nf, output_nf, ...[, ...])	Multi-Channel Attention Layer
MC_E_GCL(args, input_nf, output_nf, ...[, ...])	Multi-Channel E(n) Equivariant Convolutional Layer
MC_E_GCL_Prompt(args, input_nf, output_nf, ...)	Multi-Channel E(n) Equivariant Convolutional Layer