nn.CosineCutoff

class nn.CosineCutoff(*args: Any, **kwargs: Any)

Behler-style cosine cutoff module.

\[\begin{split}f(r) = \begin{cases} 0.5 \times \left[1 + \cos\left(\frac{\pi r}{r_\text{cutoff}}\right)\right] & r < r_\text{cutoff} \\ 0 & r \geqslant r_\text{cutoff} \\ \end{cases}\end{split}\]

Parameters:

cutoff (float, optional) – cutoff radius.