pygmtools.pytorch_backend.gamgm¶

pygmtools.pytorch_backend.gamgm(A, W, ns, n_univ, U0, init_tau, min_tau, sk_gamma, sk_iter, max_iter, quad_weight, converge_thresh, outlier_thresh, bb_smooth, verbose, cluster_M=None, projector='sinkhorn', hung_iter=True)[source]¶: Pytorch implementation of Graduated Assignment for Multi-Graph Matching (with compatibility for 2GM and clustering)