nn.GaussianRBF

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

Gaussian radial basis functions.

Parameters:

  • n_rbf – total number of Gaussian functions, \(N_g\).

  • cutoff – center of last Gaussian function, \(\mu_{N_g}\)

  • start – center of first Gaussian function, \(\mu_0\).

  • trainable – If True, widths and offset of Gaussian functions are adjusted during training process.