code: def radius_gaussian(sq_r, sig, eps=1e-9):
      """
      Compute a radius gaussian (gaussian of distance)
      :param sq_r: input radiuses [dn, ..., d1, d0]
      :param sig: extents of gaussians [d1, d0] or [d0] or float
      :return: gaussian of sq_r [dn, ..., d1, d0]
      """
      return torch.exp(-sq_r / (2 * sig**2 + eps))
  
imports: import torch
  from torch_points3d.core.common_modules.gathering import gather
label_desc: Use image node_id 1 for calling a global function with example usage: radius_gaussian(sq_r, sig, eps) and returns: torch