pygmtools.pytorch_backend.sm

pygmtools.pytorch_backend.sm(K: torch.Tensor, n1: torch.Tensor, n2: torch.Tensor, n1max, n2max, x0: torch.Tensor, max_iter: int) → torch.Tensor

Pytorch implementation of SM algorithm.