nn.GaussianRBFCentered

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

Gaussian radial basis functions centered at the origin.

Parameters:

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

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

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

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