core.models.equiformer_v2.radial_function

core.models.equiformer_v2.radial_function#

Classes#

RadialFunction

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

Module Contents#

class core.models.equiformer_v2.radial_function.RadialFunction(channels_list)#

Bases: torch.nn.Module

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

net#
forward(inputs)#