nfp.layers

Keras layers that handle the node, edge, and connectivity features returned by the preprocessor classes.

Functions

batched_segment_op

Flattens data and segment_ids containing a batch dimension for tf.math.segment* operations.

Classes

ConcatDense	Layer to combine the concatenation and two dense layers.
Gather
RBFExpansion	Layer to calculate radial basis function 'embeddings' for a continuous input variable.
Reduce
Tile