MolecularDiffusion.modules.layers.equiformer_v2_s.radial_function

Classes

RadialFunction

Contruct a radial function (linear layers + layer normalization + SiLU) given a list of channels

Module Contents

class MolecularDiffusion.modules.layers.equiformer_v2_s.radial_function.RadialFunction(channels_list)

Bases: torch.nn.Module

Contruct a radial function (linear layers + layer normalization + SiLU) given a list of channels

forward(inputs)

net