Method sample_rays

mapping/src/frame.py:89–91

(self, N_rays)

    @torch.no_grad()
    def sample_rays(self, N_rays):
        self.sample_mask = sample_rays(
            torch.ones_like(self.depth)[None, ...], N_rays)[0, ...]