Function _pad_cumsum

eval_code/recons/models/moge/utils/alignment.py:44–45

(cumsum: torch.Tensor)

def _pad_cumsum(cumsum: torch.Tensor):
    return torch.cat([torch.zeros_like(cumsum[..., :1]), cumsum, cumsum[..., -1:]], dim=-1)


def _compute_residual(a: torch.Tensor, xyw: torch.Tensor, trunc: float):