pygmtools.pytorch_backend.rrwm

pygmtools.pytorch_backend.rrwm(K: torch.Tensor, n1: torch.Tensor, n2: torch.Tensor, n1max, n2max, x0: torch.Tensor, max_iter: int, sk_iter: int, alpha: float, beta: float) → torch.Tensor

Pytorch implementation of RRWM algorithm.