d08f673481a2cfaa0ce702e80cc22cdd25e600e5,torch_geometric/nn/models/schnet.py,GaussianSmearing,forward,#GaussianSmearing#Any#,105

Before Change

        self.register_buffer("offsets", offsets)

    def forward(self, distances):
        coeff = -0.5 / torch.pow(self.widths, 2)
        gauss = torch.exp(coeff * torch.pow(distances - self.offsets, 2))
        return gauss


class ShiftedSoftplus(nn.Module):
    def __init__(self, beta=1, threshhold=20, shift=None):

After Change

        self.register_buffer("offset", offset)

    def forward(self, dist):
        dist = dist.view(-1, 1) - self.offset.view(1, -1)
        return torch.exp(self.coeff * torch.pow(dist, 2))


class ShiftedSoftplus(torch.nn.Module):
    def __init__(self):

Instances