MolecularDiffusion.modules.layers.equiformer_v2.gaussian_rbf¶

Copyright (c) Meta Platforms, Inc. and affiliates.

Classes¶

GaussianRadialBasisLayer

Functions¶

gaussian(x, mean, std)

Module Contents¶

class MolecularDiffusion.modules.layers.equiformer_v2.gaussian_rbf.GaussianRadialBasisLayer(num_basis, cutoff)¶

Bases: torch.nn.Module

extra_repr()¶

forward(dist, node_atom=None, edge_src=None, edge_dst=None)¶

bias¶

cutoff¶

mean¶

mean_init_max = 1.0¶

mean_init_min = 0¶

num_basis¶

std¶

std_init_max = 1.0¶

std_init_min¶

weight¶

MolecularDiffusion.modules.layers.equiformer_v2.gaussian_rbf.gaussian(x, mean, std)¶