pygmtools.pytorch_backend.gaussian_aff_fn

pygmtools.pytorch_backend.gaussian_aff_fn(feat1, feat2, sigma)[source]

Pytorch implementation of Gaussian affinity function