nn.MollifierCutoff

class nn.MollifierCutoff(*args: Any, **kwargs: Any)[source]

Mollifier cutoff module scaled to have a value of 1 at \(r=0\).

\[\begin{split}f(r) = \begin{cases} \exp\left(1 - \frac{1}{1 - \left(\frac{r}{r_\text{cutoff}}\right)^2}\right) & r < r_\text{cutoff} \\ 0 & r \geqslant r_\text{cutoff} \\ \end{cases}\end{split}\]
Parameters:
  • cutoff – Cutoff radius.

  • eps – Offset added to distances for numerical stability.