PeriodicRadiusGraph

PeriodicRadiusGraph#

class nfflr.nn.transform.PeriodicRadiusGraph(cutoff: float = 5.0, dtype=torch.float32)[source]#

Periodic radius graph transform.

Methods

forward(x)

Compute periodic radius graph.

forward(x: Atoms)[source]#

Compute periodic radius graph.