code: def forward(self, dist):
      dist = dist.view(-1, 1) - self.offset.view(1, -1)
      return torch.exp(self.coeff * torch.pow(dist, 2))
  
imports: from math import pi
  import torch
  import torch.nn.functional
  from torch.nn import Embedding, Sequential, Linear
  from torch_scatter import scatter
  from torch_geometric.nn import radius_graph
label_desc: Use image node_id 2 for calling the emb obj's underlying member method code with example usage: obj.forward(dist) and returns: torch