tfep.nn.embeddings.radial

Encoders to build input features.

Functions

behler_parrinello_cosine_switching_function(...)

Compute the value of the Behler-Parrinello switching function.

Classes

BehlerParrinelloRadialExpansion(r_cutoff, ...)	Expands distance into a soft one-hot encoded vector using a Gaussian basis with a cosine switching function.
GaussianBasisExpansion(means, stds[, ...])	Expands a float into a soft one-hot encoded vector using a Gaussian basis.