pygmtools.pytorch_backend.inner_prod_aff_fn

pygmtools.pytorch_backend.inner_prod_aff_fn(feat1, feat2)[source]

Pytorch implementation of inner product affinity function