MolecularDiffusion.modules.layers.esen.nn.gaussian_embedding¶

Classes¶

GaussianFeatureEmbedding

Module Contents¶

class MolecularDiffusion.modules.layers.esen.nn.gaussian_embedding.GaussianFeatureEmbedding(num_channels, num_basis=20, start=0.0, stop=10.0, trainable=False)¶

Bases: torch.nn.Module

forward(x)¶

num_basis = 20¶

num_channels¶

trainable = False¶