RBFExpansion

RBFExpansion#

class nfflr.nn.RBFExpansion(vmin: float = 0, vmax: float = 8, bins: int = 40, lengthscale: float | None = None)[source]#

Expand interatomic distances with radial basis functions.

Methods

forward(distance)

Apply RBF expansion to interatomic distance tensor.

forward(distance: Tensor) Tensor[source]#

Apply RBF expansion to interatomic distance tensor.