diffalign.models.encoder.egnn

Functions

unsorted_segment_sum(data, segment_ids, ...)

TensorFlow-style unsorted_segment_sum for 2D tensors [E, C].

Classes

EGNN(in_node_nf, in_edge_nf, hidden_nf[, ...])
E_GCL(input_nf, output_nf, hidden_nf[, ...])