Cosine

class nfflr.nn.Cosine(r_cutoff: float = 4)

Cosine cutoff profile.

\[f(r) = 0.5 \left(1 + \cos(\pi \frac{r}{r_{cut}})\right)\]

Parameters:

r_cutofffloat

cutoff radius

Examples

rs = torch.linspace(0, 5, 100)
cut = nfflr.nn.Cosine(4.0)
plt.plot(rs, cut(rs))
plt.xlabel("r")
plt.ylabel("$f_{cut}$")
plt.xlim(0, 5)
plt.show()

Methods

forward(r)