Method cat

pointersect/inference/structures.py:1092–1100

(rays: T.List['Ray'], dim: int)

    @staticmethod
    def cat(rays: T.List['Ray'], dim: int) -> 'Ray':
        out_dict = dict()
        for name in ['origins_w', 'directions_w']:
            arr = [getattr(p, name, None) for p in rays]
            if None in arr:
                out_dict[name] = None
            else:
                out_dict[name] = torch.cat(arr, dim=dim)
        return Ray(**out_dict)

    def masked_fill(self, mask: torch.Tensor, ray_src: 'Ray'):
        self.origins_w[mask] = ray_src.origins_w[mask]